python - 查找二维数组中具有某些非零元素的列的索引

Question

我有一个带有暗淡(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])

是并非所有行都为零的列索引(恰好与您给出的示例相同)。