numpy庫中的sum函數用于計算數組中元素的和。
用法:
numpy.sum(a, axis=None, dtype=None, out=None, keepdims=
參數說明:
返回值: 返回一個數組或者標量,表示數組中元素的和。
示例:
import numpy as np
a = np.array([1, 2, 3, 4, 5])
print(np.sum(a)) # 輸出15
b = np.array([[1, 2, 3], [4, 5, 6]])
print(np.sum(b)) # 輸出21
c = np.array([[1, 2, 3], [4, 5, 6]])
print(np.sum(c, axis=0)) # 輸出[5 7 9]
d = np.array([[1, 2, 3], [4, 5, 6]])
print(np.sum(d, axis=1)) # 輸出[ 6 15]
e = np.array([[1, 2, 3], [4, 5, 6]])
print(np.sum(e, axis=1, keepdims=True)) # 輸出[[ 6] [15]]
上述代碼中,分別計算了數組a、b、c、d、e中元素的和。