Python numpy 数组计算

Question

我使用 numpy histogram by N,a = np.histogram(z,bins=50) 计算了一个直方图切片。现在我的a 包含z 的50 个切片的值，N 包含这些切片内的计数。

我想为a 计算R-r

我试过了

result = [R-r for R,r in zip(a[1:],a)]

但它似乎不起作用。我在这里做错了什么？

Answer 1

您只需为此使用numpy.diff：

>>> a = np.array([1,3,6,8,17,23,45])
>>> np.diff(a)
array([ 2,  3,  2,  9,  6, 22])

编辑：

您的代码也可以正常工作，但您不应为此使用列表推导，因为 NumPy 已经为此提供了一个函数，因为它既快速又高效。

>>> a = np.array([1,3,6,8,17,23,45])
>>> [R-r for R,r in zip(a[1:],a)]
[2, 3, 2, 9, 6, 22]