必须有一种(非常)快速有效的方法来仅从 numpy 数组中获取元素,或者更有趣的是从它的切片中获取元素。 假设我有一个 numpy 数组:
import numpy as np
a = np.arange(-10,10)
如果我有一个列表:
s = [9, 12, 13, 14]
我可以从 a 中选择元素:
a[s] #array([-1, 2, 3, 4])
我怎样才能拥有一个由 a[s] 中满足条件的元素组成的(numpy)数组,即是正数(或负数)? 结果应该是
np.ifcondition(a[s]>0, a[s]) #array([2, 3, 4])
它看起来微不足道,但我无法找到一个简单而浓缩的表达方式。我敢肯定口罩可以,但对我来说看起来并不直接。 但是,两者都不是:
a[a[s]>0]
a[s[a[s]>0]]
其实都是不错的选择。
最佳答案
怎么样:
In [19]: b = a[s]
In [20]: b[b > 0]
Out[20]: array([2, 3, 4])
关于python - 仅获取满足 numpy 数组中条件的值,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/13029029/