python - 关于代码条件

Question

我有一个数组数据集。我想取出那些值位于(0.5，0.6)的数组。

我之前做了不到 0.4

代码是

c_subset = np.array(c)[np.where(np.array(c)<0.4)]

任何人都可以帮助如何将上面的行修改为间隔条件？

Answer 1

使用 numpy logical and :

>>> c = [0.50, 0.52, 0.54, 0.56, 0.58, 0.60]
>>> a = np.array(c)
>>> a[(0.5 < a) & (a < 0.6)]
array([0.52, 0.54, 0.56,  0.58])

请注意 numpy.where没有必要。