有人可以帮助我有条件地选择 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)
Results:
[ 0.725 0.99 ]
[ 0.725 0.99 ]
[ 0.725 0.99 ]
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
