python - 如何使用Python在数组的所有第三列中找到精确值？

Question

如果我有一个像这样的 Numpy 数组:

array([0, 0, 0]
      [0, 0, 1]
      [0, 1, 2]
      [0, 1, 3]
      [1, 0, 4]
      [1, 0, 5]
      [1, 1, 6]
      [1, 1, 7])

我尝试使用

if c==1 in range(X[:,2]):
  print 'yes'

但出现错误

TypeError: only length-1 arrays can be converted to Python scalars

是否可以使用切片和 if 语句找到 c==1？

Answer 1

因为 x[:,2] 为您提供了第三列中的元素数组

>>> import numpy as np
>>> x = np.array([[0, 0, 0], [0, 0, 1], [0, 1, 2], [0, 1, 3], [1, 0, 4], [1, 0, 5], [1, 1, 6], [1, 1, 7]])
>>> x
array([[0, 0, 0],
       [0, 0, 1],
       [0, 1, 2],
       [0, 1, 3],
       [1, 0, 4],
       [1, 0, 5],
       [1, 1, 6],
       [1, 1, 7]])
>>> x[:,2]
array([0, 1, 2, 3, 4, 5, 6, 7])
>>> if 1 in x[:,2]:
...     print("yes")
... 
yes

这样效率最高

>>> if np.any(x[:, 2] == 1):
...     print("yes")
... 
yes
>>>