轉自:https://blog.csdn.net/Muzi_Water/article/details/85104941
mean和average都是計算均值的函數,在不指定權重的時候average和mean是一樣的。指定權重后,average可以計算一維的加權平均值。具體如下:
import numpy as np a = np.array([np.random.randint(0, 20, 5), np.random.randint(0, 20, 5)]) print('原始數據\n', a) print('mean函數'.center(20, '*')) print('對所有數據計算\n', a.mean()) print('axis=0,按行方向計算,即每列\n', a.mean(axis=0)) # 按行方向計算,即每列 print('axis=1,按列方向計算,即每行\n', a.mean(axis=1)) # 按列方向計算,即每行 print('average函數'.center(20, '*')) print('對所有數據計算\n', np.average(a)) print('axis=0,按行方向計算,即每列\n', np.average(a, axis=0)) # 按行方向計算,即每列 print('axis=1,按列方向計算,即每行\n', np.average(a, axis=1)) # 按列方向計算,即每行 b = np.array([1, 2, 3, 4]) wts = np.array([4, 3, 2, 1]) print('不指定權重\n', np.average(b)) print('指定權重\n', np.average(b, weights=wts))
運行結果:
原始數據 [[10 12 7 14 5] [12 10 2 16 7]] *******mean函數******* 對所有數據計算 9.5 axis=0,按行方向計算,即每列 [ 11. 11. 4.5 15. 6. ] axis=1,按列方向計算,即每行 [ 9.6 9.4] *****average函數****** 對所有數據計算 9.5 axis=0,按行方向計算,即每列 [ 11. 11. 4.5 15. 6. ] axis=1,按列方向計算,即每行 [ 9.6 9.4] 不指定權重 2.5 指定權重 2.0