在 Numpy 陣列中尋找最近值
在 numpy 中,可以透過簡單的過程找到給定目標值的最近值。讓我們定義一個函數來滿足此要求:
import numpy as np def find_nearest(array, value): array = np.asarray(array) idx = (np.abs(array - value)).argmin() return array[idx]
函數接受一個陣列和一個目標值。它將陣列轉換為 numpy 數組,並計算數組每個元素與目標值之間的絕對差。使用 argmin() 決定具有最小絕對差的元素的索引。最後,傳回該索引處的元素。
用法範例:
讓我們用一個範例來說明find_nearest() 函數的用法:
array = np.random.random(10) print(array) print(find_nearest(array, value=0.5))
在此範例中,建立了一個包含10 個元素的隨機數組。然後使用 find_nearest() 函數尋找數組中最接近 0.5 的元素。輸出將是數組中最接近 0.5 的值。
以上是如何找出 NumPy 數組中最接近的值?的詳細內容。更多資訊請關注PHP中文網其他相關文章!