arrays - 使用花式索引从 Numpy 数组中查找和删除全零列

Question

如何在 numpy 数组中找到全为零的列，然后从数组中删除它们？我正在寻找一种方法来获取列索引，然后使用这些索引进行删除。

Answer 1

您可以使用 np.argwhere , 与 np.all找到你的指数。要删除它们，请使用 np.delete .

示例 :

找到您的 0列:

a = np.array([[1, 2, 0, 3, 0],
              [4, 5, 0, 6, 0],
              [7, 8, 0, 9, 0]])

idx = np.argwhere(np.all(a[..., :] == 0, axis=0))

>>> idx
array([[2],
       [4]])

删除您的列

a2 = np.delete(a, idx, axis=1)

>>> a2
array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])