## NumPy的应用-2
### 数组对象的方法
#### 获取描述统计信息
描述统计信息主要包括数据的集中趋势、离散程度和频数分析等,其中集中趋势主要看均值和中位数,离散程度可以看极值、方差、标准差等,详细的内容大家可以阅读[《统计思维系列课程01:解读数据》](https://zhuanlan.zhihu.com/p/595273755)。
```Python
array1 = np.random.randint(1, 100, 10)
array1
```
输出:
```
array([46, 51, 15, 42, 53, 71, 20, 62, 6, 94])
```
**计算总和、均值和中位数。**
代码:
```Python
print(array1.sum())
print(np.sum(array1))
print(array1.mean())
print(np.mean(array1))
print(np.median(array1))
print(np.quantile(array1, 0.5))
```
> **说明**:上面代码中的`mean`、`median`和`quantile`分别是 NumPy 中计算算术平均值、中位数和分位数的函数,其中`quantitle`函数的第二个参数设置为0.5表示计算50%分位数,也就是中位数。
输出:
```
460
460
46.0
46.0
48.5
48.5
```
**极值、全距和四分位距离。**
代码:
```Python
print(array1.max())
print(np.amax(array1))
print(array1.min())
print(np.amin(array1))
print(array1.ptp())
print(np.ptp(array1))
q1, q3 = np.quantile(array1, [0.25, 0.75])
print(q3 - q1)
```
输出:
```
94
94
6
6
88
88
34.25
```
**方差、标准差和变异系数。**
代码:
```Python
print(array1.var())
print(np.var(array1))
print(array1.std())
print(np.std(array1))
print(array1.std() / array1.mean())
```
输出:
```
651.2
651.2
25.51862065237853
25.51862065237853
0.5547526228777941
```
**绘制箱线图。**
箱线图又称为盒须图,是显示一组数据分散情况的统计图,因形状如箱子而得名。 它主要用于反映原始数据分布的特征,还可以进行多组数据分布特征的比较。
代码:
```Python
plt.boxplot(array1, showmeans=True)
plt.ylim([-20, 120])
plt.show()
```
输出:
值得注意的是,对于二维或更高维的数组,在获取描述统计信息时,可以通过名为`axis`的参数指定均值、方差等运算是沿着哪一个轴来执行,`axis`参数不同,执行的结果可能是大相径庭的,如下所示。
代码:
```Python
array2 = np.random.randint(60, 101, (5, 3))
array2
```
输出:
```
array([[72, 64, 73],
[61, 73, 61],
[76, 85, 77],
[97, 88, 90],
[63, 93, 82]])
```
代码:
```Python
array2.mean()
```
输出:
```
77.0
```
代码:
```Python
array2.mean(axis=0)
```
输出:
```
array([73.8, 80.6, 76.6])
```
代码:
```Python
array2.mean(axis=1)
```
输出:
```
array([69.66666667, 65. , 79.33333333, 91.66666667, 79.33333333])
```
代码:
```Python
array2.max(axis=0)
```
输出:
```
array([97, 93, 90])
```
代码:
```Python
array2.max(axis=1)
```
输出:
```
array([73, 73, 85, 97, 93])
```
再看看绘制箱线图,对于二维数组每一列都会产生一个统计图形,如下所示。
代码:
```Python
plt.boxplot(array2, showmeans=True)
plt.ylim([-20, 120])
plt.show()
```
输出:
> **说明**:箱线图中的小圆圈用来表示离群点,也就是大于$\small{Q_3 + 1.5 \times IQR}$或小于$\small{Q_1 - 1.5 \times IQR}$的值。公式中的常量`1.5`可以通过绘制箱线图的`boxplot`函数的`whis`参数进行修改,常用的值是`1.5`和`3`,修改为`3`通常是为了标识出极度离群点。
需要说明的是,NumPy 的数组对象并没有提供计算几何平均值、调和平均值、去尾平均值等的方法,如果有这方面的需求,可以使用名为 scipy 的三方库,它的`stats`模块中提供了这些函数。此外,该模块还提供了计算众数、变异系数、偏态、峰度的函数,代码如下所示。
代码:
```python
from scipy import stats
print(np.mean(array1)) # 算术平均值
print(stats.gmean(array1)) # 几何平均值
print(stats.hmean(array1)) # 调和平均值
print(stats.tmean(array1, [10, 90])) # 去尾平均值
print(stats.variation(array1)) # 变异系数
print(stats.skew(array1)) # 偏态系数
print(stats.kurtosis(array1)) # 峰度系数
```
输出:
```
46.0
36.22349548825599
24.497219530825497
45.0
0.5547526228777941
0.11644192634527782
-0.7106251396024126
```
#### 其他相关方法概述
1. `all()` / `any()`方法:判断数组是否所有元素都是`True` / 判断数组是否有为`True`的元素。
2. `astype()`方法:拷贝数组,并将数组中的元素转换为指定的类型。
3. `reshape()`方法:调整数组对象的形状。
4. `dump()`方法:保存数组到二进制文件中,可以通过 NumPy 中的`load()`函数从保存的文件中加载数据创建数组。
代码:
```Python
array.dump('array1-data')
array3 = np.load('array1-data', allow_pickle=True)
array3
```
输出:
```
array([46, 51, 15, 42, 53, 71, 20, 62, 6, 94])
```
5. `tofile()`方法:将数组对象写入文件中。
```Python
array1.tofile('res/array.txt', sep=',')
```
6. `fill()`方法:向数组中填充指定的元素。
7. `flatten()`方法:将多维数组扁平化为一维数组。
代码:
```Python
array2.flatten()
```
输出:
```
array([1, 2, 3, 4, 5, 6, 7, 8, 9])
```
8. `nonzero()`方法:返回非0元素的索引。
9. `round()`方法:对数组中的元素做四舍五入操作。
10. `sort()`方法:对数组进行就地排序。
代码:
```Python
array1.sort()
array1
```
输出:
```
array([ 6, 15, 20, 42, 46, 51, 53, 62, 71, 94])
```
11. `swapaxes()`和`transpose()`方法:交换数组指定的轴和转置。
代码:
```Python
array2.swapaxes(0, 1)
```
输出:
```
array([[1, 4, 7],
[2, 5, 8],
[3, 6, 9]])
```
代码:
```Python
array2.transpose()
```
输出:
```
array([[1, 4, 7],
[2, 5, 8],
[3, 6, 9]])
```
12. `tolist()`方法:将数组转成 Python 中的`list`。
代码:
```Python
print(array2.tolist())
print(type(array2.tolist()))
```
输出:
```
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
```