2次元配列内の非ゼロ要素を持つ列のインデックスを検索

Question

dim（157,1944）のnumpy配列があります。

任意の行に非ゼロ要素を持つ列のインデックスを取得したいとします。

例：[[0,0,3,4], [0,0,1,1]] ---->[2,3]

あなたはそれぞれの行を見ると、列の非ゼロ要素がある[2、3]私が持っているのであれば

[[0,1,3,4], [0,0,1,1]]

[1,2,3]を取得する必要があります。これは、列インデックス0に任意の行の非ゼロ要素がないためです。

Answer 1

質問が完全に定義されているかどうかわかりません。しかし、我々は行のなしがゼロであるため、列のインデックスがある

import numpy as np 
a = np.array([[0,0,3,4], [0,0,1,1]]) 

その後、

>>> np.nonzero(np.all(a != 0, axis=0))[0] 
array([2, 3]) 

で始まると言う、と

>>> np.nonzero(np.any(a != 0, axis=0))[0] 
array([2, 3]) 

はの指標であります のすべての列がではない列はゼロです（これは、指定した例でも同じです）。