Python 求和两个数字之间的元素

Question

我相信这很简单。我搜索了谷歌和这里，没有找到具体的答案

a = rnd.randn(100)
print np.sum(a)

给出a中元素的总和

np.sum(a[a>0.])

给出大于0的元素之和

print np.sum((a < 2.0) & (a > -2.0))

好的，所以这将返回 2 到 -2 之间的元素数。

如何得到 2 和 -2 之间的元素之和？？？例如，我已经尝试了很多东西

np.sum(a[a >0.] & a[a<1.])

等等，找不到正确的方法:-(

Answer 1

& 是按位运算符，不会为您提供正确的结果，您需要使用 np.logical_and 来获取掩码数组。然后你可以将它作为索引传递给数组以获得想要的项目，然后将它传递给 sum:

In [9]: a = np.arange(-10, 10)

In [10]: a[np.logical_and(a>-2,a<2)]
Out[10]: array([-1,  0,  1])

In [11]: a[np.logical_and(a>-2,a<2)].sum()
Out[11]: 0