numpy.sum()函数

点击查看代码

# 当axis=0，就是将一个矩阵的每一列向量相加
# 当axis=1，就是将一个矩阵的每一行向量相加
# 当没有axis时，只是单纯地将所有元素相加

import numpy as np

A = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
# print(np.sum(A, axis=1))    # output [ 6 15 24]
# print(np.sum(A, axis=0))  # output [12 15 18]
print(np.sum(A))  # output 45