NumPy库中的mean()函数是一个用于求取数组平均值的函数,它可以计算多维数组的平均值,也可以计算指定轴的平均值。mean()函数的使用方法如下:
1. 计算数组的平均值
import numpy as np a = np.array([[1,2,3], [4,5,6]]) print(np.mean(a))
上面的代码会输出3.5,即数组a的平均值。
2. 计算指定轴的平均值
import numpy as np a = np.array([[1,2,3], [4,5,6]]) print(np.mean(a, axis=0))
上面的代码会输出[2.5 3.5 4.5],即数组a沿着第0轴的平均值。
3. 计算多维数组的平均值
import numpy as np a = np.array([[[1,2,3], [4,5,6]], [[7,8,9], [10,11,12]]]) print(np.mean(a))
上面的代码会输出6.5,即多维数组a的平均值。
4. 计算多维数组指定轴的平均值
import numpy as np a = np.array([[[1,2,3], [4,5,6]], [[7,8,9], [10,11,12]]]) print(np.mean(a, axis=1))
上面的代码会输出[[2.5 3.5 4.5] [8.5 9.5 10.5]],即多维数组a沿着第1轴的平均值。
5. 计算多维数组指定轴的平均值(使用keepdims参数)
import numpy as np a = np.array([[[1,2,3], [4,5,6]], [[7,8,9], [10,11,12]]]) print(np.mean(a, axis=1, keepdims=True))
上面的代码会输出[[[2.5 3.5 4.5]] [[8.5 9.5 10.5]]],即多维数组a沿着第1轴的平均值,使用keepdims参数保持原有的维度。
6. 使用out参数
import numpy as np a = np.array([[[1,2,3], [4,5,6]], [[7,8,9], [10,11,12]]]) b = np.zeros((2,3)) print(np.mean(a, axis=1, out=b)) print(b)
上面的代码会输出[[2.5 3.5 4.5] [8.5 9.5 10.5]],即多维数组a沿着第1轴的平均值,使用out参数将结果输出到b中。
7. 使用ddof参数
import numpy as np a = np.array([[1,2,3], [4,5,6]]) print(np.mean(a, axis=0, ddof=1))
上面的代码会输出[2.33333333 3.33333333 4.33333333],即数组a沿着第0轴的平均值,使用ddof参数改变自由度。
NumPy库中的mean()函数可以用来计算多维数组的平均值,也可以计算指定轴的平均值,可以使用keepdims参数保持原有的维度,也可以使用out参数将结果输出到指定数组,还可以使用ddof参数改变自由度。