如何在python中选择ndarray数据？

Question

我有一个ndarray类型的数组，例如：

1, 2, 0.5
2, 6, 0.9
9, 2, 0.83

我想保留第3个元素大于0.8的行，并丢弃其他行。这意味着我想要这个结果：

2, 6, 0.9
9, 2, 0.83

我该如何完成任务？

Answer 1

以下是您问题的简单实现：

import numpy as np
data = np.array([[1, 2, 0.5],[2, 6, 0.9],[9, 2, 0.83]])
result =data[data[:,2]>0.8]

输出：

[[ 2.    6.    0.9 ]
[ 9.    2.    0.83]]