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]]