[Python]Numpy配列の2番目に小さい値を取得する（array get second smallest value）には？

どうも、ちょげ(@chogetarou)です。

Numpy配列(array)の2番目の最小値を取得する方法を紹介します。

方法

Numpy配列(array)で2番目に小さい値を取得するには、partition()を使います。

まず、numpyからpartition()を呼び出します。

partition()の第１引数にnumpyから呼び出したunique()、第２引数に「1」を指定します。

unique()の引数に、Numpy配列から呼び出したflatten()の結果を指定します。

そして、partition()の結果のインデックス「1」にアクセスします。

#arr=対象のNumpy配列
result = np.partition(np.unique(arr.flatten(), 1)[1] #２番目に小さい値を取得

上記のpartition()のインデックス「1」にアクセスすることで、Numpy配列の2番目に小さい値を取得できます。

import numpy as np

nums = np.array([[10, 3, 1, 5, 8], [1, 7, 1, 2, 9]])

result = np.partition(np.unique(nums.flatten()), 1)[1]

print(result)

出力：
2