Skip to main content
 首页 » 编程设计

python之numpy 数组中元素 i 和 i+1 的平均值

2025年05月04日12mq0036

出于好奇,是否有一个特定的 numpy 函数可以执行以下操作(据说会更快):

a = np.array((0,2,4)) 
b = np.zeros(len(a) - 1) 
for i in range(len(b)): 
    b[i] = a[i:i+2].mean() 
 
print(b) 
#prints [1,3] 

干杯

请您参考如下方法:

你可以使用

b = (a[1:] + a[:-1]) / 2. 

避免 Python 循环。