数组统计运算
通过NumPy库中的相关方法,我们可以很方便地运用Python进行数组的统计汇总,比如计算数组极大值、极小值以及平均值等。表1列举了NumPy数组中与统计运算相关的方法。
表1 NumPy数组中与统计运算相关的方法
方法 | 描述 |
---|---|
sum | 对数组中全部或某个轴向的元素求和 |
mean | 算术平均值 |
min | 计算数组中的最小值 |
max | 计算数组中的最大值 |
argmin | 表示最小值的索引 |
argmax | 表示最大值的索引 |
cumsum | 所有元素的累计和 |
cumprod | 所有元素的累计积 |
需要注意的是,当使用ndarray对象调用cumsum()和cumprod()方法后,产生的结果是一个由中间结果组成的数组。
为了能让读者更好地理解,接下来,通过一些示例来演示上述方法的具体用法,代码如下。
In [103]: arr = np.arange(10)
In [104]: arr.sum() # 求和
Out[104]: 45
In [105]: arr.mean() # 求平均值
Out[105]: 4.5
In [106]: arr.min() # 求最小值
Out[106]: 0
In [107]: arr.max() # 求最大值
Out[107]: 9
In [108]: arr.argmin() # 求最小值的索引
Out[108]: 0
In [109]: arr.argmax() # 求最大值的索引
Out[109]: 9
In [110]: arr.cumsum() # 计算元素的累计和
Out[110]: array([ 0, 1, 3, 6, 10, 15, 21, 28, 36, 45])
In [111]: arr.cumprod() # 计算元素的累计积
Out[111]: array([0, 0, 0, 0, 0, 0, 0, 0, 0, 0])