Python 如何在numpy数组中有条件地选择元素

有人能帮我有条件地选择numpy数组中的元素吗？我试图返回大于阈值的元素。我目前的解决办法是：

sampleArr = np.array([ 0.725, 0.39, 0.99 ])
condition = (sampleArr > 0.5)`
extracted = np.extract(condition, sampleArr) #returns [0.725 0.99]

---

然而，这似乎是迂回的，我怀疑有一种方法可以在一行中完成它？

您可以像以下那样直接索引：

sampleArr[sampleArr > 0.5]

测试代码： 结果：

---

实际上，您可以这样做布尔索引：

extracted = sampleArr[sampleArr > 0.5]

---

谢谢，我修好了。看起来@stephernauch无论如何都比我强：）你的答案是一样的：P

[ 0.725  0.99 ]
[ 0.725  0.99 ]
[ 0.725  0.99 ]

extracted = sampleArr[sampleArr > 0.5]