numpy.sum()可接受的参数是:
sum(a, axis, dtype, out, keepdims)
- 参数 a
要进行加法运算的向量/数组/矩阵 - 参数 axis
=None(默认值)时,将数组/矩阵中的元素全部相加;
=0 时,将每一列的元素相加,将矩阵压缩为一行;
=1 时,将每一行的元素相加,将矩阵压缩为一列;
=元组时,按axis=元组中各元素分别相加再求和;
注意,axis 的取值不能超过数组/矩阵的维度。
a = np.array([[1,1,1],[2,2,2],[3,3,3]])
print(a):
[[1 1 1]
[2 2 2]
[3 3 3]]
print(np.sum(a))
把整个数组a加起来:18
print(np.sum(a,axis=0))
把数组a按列相加:[6 6 6]
print(np.sum(a,axis=1))
把数组a按行相加:[3 6 9]
print(np.sum(y,axis=(0,1)))
把数组a按axis=0和axis=1分别相加再求和:18
- 如果两个 list 相加,结果是把两个 list 连成了一个list。如果是两个numpy.ndarray 相加,结果是把每个数组对应的维度相加。例如:
a = [1,2]
b = [3,4]
print(a+b):[1, 2, 3, 4]
c = np.array([1,2])
d = np.array([3,4])
print(c+d):[4 6]