NumPy 配列で最も近い値を見つけるにはどうすればよいですか?

NumPy 配列内の最も近い値の特定

NumPy 配列内の指定された値に最も近い要素を特定することは、一般的なタスクとなる場合があります。 np.find_nearest() 関数は、そのような値を見つけるための便利な方法を提供します。

カスタム関数のアプローチ

次に、find_nearest() 関数のカスタム NumPy 実装を示します。

import numpy as np

def find_nearest(array, value):
    array = np.asarray(array)
    idx = (np.abs(array - value)).argmin()
    return array[idx]

この関数は引数として配列とターゲット値を受け取ります。 NumPy の np.abs() 関数を利用して、配列内の各要素とターゲット値の間の絶対差を計算します。次に、argmin() 関数を使用して、絶対差の最小値のインデックス位置を特定します。

使用例

次の NumPy 配列を考えてみましょう。

array = np.random.random(10)
print(array)
# [ 0.21069679  0.61290182  0.63425412  0.84635244  0.91599191  0.00213826
#   0.17104965  0.56874386  0.57319379  0.28719469]

0.5 に最も近い値を検索するには、find_nearest() を呼び出します。 function:

print(find_nearest(array, value=0.5))
# 0.568743859261

この関数は、ターゲット値 0.5 との差が最も小さい配列内の要素を正確に識別します。

以上がNumPy 配列で最も近い値を見つけるにはどうすればよいですか?の詳細内容です。詳細については、PHP 中国語 Web サイトの他の関連記事を参照してください。