python之如何有条件地选择numpy数组中的元素

itcoder 阅读：86 2024-07-26 13:47:55 评论：0

有人可以帮助我有条件地选择 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]

测试代码:

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

结果:

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

标签:Python

声明

1.本站遵循行业规范，任何转载的稿件都会明确标注作者和来源；2.本站的原创文章，请转载时务必注明文章作者和来源，不尊重原创的行为我们将追究责任；3.作者投稿可能会经我们编辑修改或补充。