如何使用numpy accumulator和add函数逐列添加数组以构成基本累加器?
import numpy as np
a = np.array([1,1,1])
b = np.array([2,2,2])
c = np.array([3,3,3])
two_dim = np.array([a,b,c])
y = np.array([0,0,0])
for x in two_dim:
y = np.add.accumulate(x,axis=0,out=y)
return y
实际输出:[1,2,3]
所需的输出:[6,6,6]
numpy词汇表说sum along axis参数axis = 1对行求和:“我们可以对数组的每一行求和,在这种情况下,我们沿列或轴1进行操作”.
“二维数组具有两个相应的轴:第一个数组在行上垂直向下(轴0),第二个数组在行上垂直向下(轴1)”
当axis = 1时,我期望输出[3,6,9],但这也会返回[1,2,3].
当然! x和y都不是二维的.
我究竟做错了什么?
我可以手动使用np.add()
aa = np.array([1,1,1])
bb = np.array([2,2,2])
cc = np.array([3,3,3])
yy = np.array([0,0,0])
l = np.add(aa,yy)
m = np.add(bb,l)
n = np.add(cc,m)
print n
现在我得到正确的输出,[6,6,6]