python - 检查索引处 numpy 数组的值

Question

我最近开始使用 numpy 工作。我正在尝试测试二维数组是否包含特定的子数组。下面的代码返回一个错误。我该如何解决这个问题？

import numpy as np

testArray = np.array([[None, 0], [None, 0], [None, 0], [None, 0], [None, 0], [None, 0], [None, 0], [None, 0], [None, 0]])

for i in range(len(testArray)):
    if (testArray[i] == [None, 0]):
        print(i)

Answer 1

无需迭代，您可以使用全部:

>>> testArray[(testArray == [None,0]).all(1)]
array([[None, 0],
       [None, 0],
       [None, 0],
       [None, 0],
       [None, 0],
       [None, 0],
       [None, 0],
       [None, 0]], dtype=object)

或者，如果您只想查看该子数组是否存在，请另外使用any:

>>> (testArray == [None,0]).all(1).any()
True