Python sum()函数和.sum(axis=0)函数

sum()函数

sum()函数是Python的内置函数，用于计算数字序列（如列表、元组）的总和。如果不提供参数，它会计算序列中所有元素的总和。

import numpy as np

a = np.array([1, 2, 3, 4])
b = sum(a)
print(b)

输出：
10

.sum(axis=0)函数

.sum(axis=0)函数是NumPy库中数组对象的一个方法，用于沿着指定的轴（axis）对数组进行求和。当axis=0时，它对列进行求和。

import numpy as np

a = np.array([[1, 2, 3, 4],
              [5, 6, 7, 8],
              [11, 22, 33, 44]])
b = sum(a)  # 按列进行求和
print(b)

输出：
[17 30 43 56]

import numpy as np

a = np.array([[[1, 2, 3, 4],
              [5, 6, 7, 8],
              [11, 22, 33, 44]
              ],
             [[-1, -5, -8, 5],
              [0, 2, 4, 6],
              [6, 1, 0, 4]
              ]])
b = a.sum(axis=0)  # a中两个矩阵对应项进行求和
print(b)

输出：
[[ 0 -3 -5  9]
 [ 5  8 11 14]
 [17 23 33 48]]

.sum(axis=1)函数和.sum(axis=2)函数

import numpy as np

a = np.array([[[1, 2, 3, 4],
              [5, 6, 7, 8],
              [11, 22, 33, 44]
              ],
             [[-1, -5, -8, 5],
              [0, 2, 4, 6],
              [6, 1, 0, 4]
              ]])
b = np.sum(a, axis=1)  # a中两个矩阵按列进行求和
c = np.sum(a, axis=2)  # a中两个矩阵按行进行求和

print("b =", b)
print("c =", c)

输出：
b = [[17 30 43 56]
 [ 5 -2 -4 15]]
c = [[ 10  26 110]
 [ -9  12  11]]