非ゼロ要素を抽出するnumpy.nonzero関数の使い方

np.nonzeroは配列を引数にとります。np.ndarray.nonzeroはndarrayのメソッドとして使えます。 インデックスを返してくれるので、うまく使えば非ゼロの必要な要素だけを抽出することができます。使い方を実際のコードでみていきましょう。 In [1]: import numpy as np In [2]: a = np.random.randint(0, 10, size=20) In [3]: a # 一次元乱数配列 Out[3]: array([0, 7, 5, 8, 4, 9, 0, 2, 1, 7, 9, 0, 6, 9, 2, 6, 0, 8, 3, 9]) In [4]: np.nonzero(a) # 非ゼロ要素のインデックスを取得 Out[4]: (array([ 1, 2, 3, 4, 5, 7, 8, 9, 10, 12, 13, 14, 15, 1