出于好奇,是否有一个特定的 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 循环。
出于好奇,是否有一个特定的 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 循环。