我有一个数组,我想使用 NumPy 选择在 Python 中这些行的不同列上有某些条件的行。例如考虑这个数组:
test_array = numpy.array(([1,2,3,5],[4,5,6,7],[7,8,9,4]))
现在我想要第 1 列为 1
且第 4 列为 5
的所有行。
因此输出应为 [1,2,3,5]
,在本例中只有 1 行。
最佳答案
您可以使用以下方法:
>>> test_array[np.where(np.all(test_array[:,[0,3]]==[1,5],axis=1))]
array([[1, 2, 3, 5]])
关于python - 根据多列的条件选择行,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/31372429/