如何从 2D numpy.array 中选择，其中列 == 条件

Question

例如，我在 2D numpy.array 中有一组记录，我想选择第 3 列中的值等于 10 的所有记录。除了遍历数组并构建所选记录的列表之外，有没有办法做到这一点？

Answer 1

在这里我创建了一个 numpy 数组。

print(df)

o/p 是： array([['A'， 'car']，['B'， 'bike']，['C'， 'car']，['D'， 'truck']]， dtype=object(

现在要选择所有带有 car 的行，您只需从数组中过滤该值即可

df[df == "car"]

o/p 是： array(['car'， 'car']， dtype=object(

如果您希望将其转换为python列表，请将语句包装在list中，即

list(df[df == "car])

Answer 2

一旦我知道了"过滤器"的概念，我就搜索了更多，并在这个堆栈溢出问题中找到了我正在寻找的答案。

因此，在汽车示例中，过滤器将写为 df[df[：，1] == 'car']