gpt4 book ai didi

python - 根据多列的条件选择行

转载 作者:太空宇宙 更新时间:2023-11-03 17:34:39 24 4
gpt4 key购买 nike

我有一个数组,我想使用 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/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com