python - numpy 二维数组上的 where 操作

Question

我有一个大小为 571x24 的 numpy 数组“A”，我试图找到其中的零索引，所以我这样做:

>>>A.shape
(571L, 24L)

import numpy as np
z1 = np.where(A==0)

z1 是一个具有以下大小的元组:

>>> len(z1)
2
>>> len(z1[0])
29
>>> len(z1[1])
29

我希望创建一个与 A 大小相同的 z1。我如何实现这一目标？

编辑:我想创建 bool 数组 z1 来表示 A 中是否存在零，这样:

>>>z1.shape
    (571L, 24L)

Answer 1

您可以使用 python 中的等于运算符和 numpy 来检查这一点。示例:

>>> A = np.array([[0,2,2,1],[2,0,0,3]])
>>> A == 0
array([[ True, False, False, False],
       [False,  True,  True, False]], dtype=bool)

np.where() 执行其他操作，请参阅 documentation 。不过，可以通过使用广播的 np.where() 来实现这一点。请参阅文档。

>>> np.where(A == 0, True, False)
array([[ True, False, False, False],
       [False,  True,  True, False]], dtype=bool)