我可以使用myarray[myarray[:,1]==7]从第二个元素为7的numpy数组中选择行。如何扩展它以选择第二个元素为 7 或 9 的行？例如。类似 myarray[myarray[:,1]==7|==9] (显然这是行不通的)。
最佳答案
使用a[(a[:,1] == 7) | (a[:,1] == 9)] 例如:
In [6]: a = np.array([[4,7,8], [6,9,0], [4,4,4]])
In [7]: a[(a[:,1] == 7) | (a[:,1] == 9)]
Out[7]:
array([[4, 7, 8],
[6, 9, 0]])
另一种选择是使用numpy.logic_or
In [15]: a[np.logical_or(a[:, 1] == 7, a[:,1] == 9)]
Out[15]:
array([[4, 7, 8],
[6, 9, 0]])
关于python - 使用多个逻辑条件从 numpy 数组中按列进行选择，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/30147034/