如何有效地获取numpy矩阵中满足条件的每列的行nr？

Question

关于numpy矩阵变换的早期question之后。

np_array = np.matrix(
   [[0,0,0,0,1,0,0,0,0,0,0],
    [0,0,0,1,0,1,0,0,0,0,0],
    [0,0,0,0,0,0,0,0,0,0,0],
    [0,0,1,0,0,0,1,0,0,0,0],
    [0,0,0,0,0,0,0,0,0,0,0],
    [0,1,0,0,0,0,0,1,0,1,1],
    [0,0,0,0,0,0,0,0,0,0,0],
    [1,0,0,0,0,0,0,0,1,0,0]]
)

在没有循环的情况下，每个列获得行nr的最有效方法是什么？如果列中没有数字为1的行，则应返回该列的np.nan。 如果这个numpy矩阵是

的结果

np.array([7,5,3,1,0,1,3,5,7,5,5])

Answer 1

i,j=np.where(np_array==1)

其中i是标识行的索引列表，j是表示列

的索引列表