《利用Python进行数据分析·第2版》第四章 Numpy基础:数组和矢量计算
numpy高效处理大数组的数据原因:
numpy是在一个连续的内存块中存储数据,独立于其他python内置对象。其C语言编写的算法库可以操作内存而不必进行其他工作。比起内置序列,使用的内存更少(即时间更快,空间更少)
numpy可以在整个数组上执行复杂的计算,而不需要借助python的for循环
4.0 前提知识
数据:结构化的数据代指所有的通用数据,如表格型,多维数组,关键列,时间序列等
相关包:numpy pandas matplotlib scipy scikit-learn statsmodels
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
import seaborn as sns
import statsmodels as sm
数据规整(Munge/Munging/Wrangling):
指的是将非结构化和(或)散乱数据处理为结构化或整洁形式的整个过程。这几个词已经悄悄成为当今数据黑客们的行话了。Munge这个词跟Lunge押韵。
伪码(Pseudocode):
算法或过程的“代码式”描述,而这些代码本身并不是实际有效的源代码。
语法糖(Syntactic sugar):
这是一种编程语法,它并不会带来新的特性,但却能使代码更易读、更易写。
4.1 Numpy的ndarrray:一种多维数组对象
ndarray:N维数组对象,快速而灵活的大数据集容器
- 通用的同构数据多维容器(所有元素相同类型)
- 每个对象都有一个shape(表示各维度大小的元组)+dtype(表示数组数据类型的对象)+ ndim(维度)
对整块数据执行运算的语法=标量元素执行运算的语法
np.random.randn(n,m):生成2行3列的正态分布随机数组
data = np.random.randn(2, 3)
array([[-0.2047, 0.4789, -0.5194],
[-0.5557, 1.9658, 1.3934]])
data * 10
array([[ -2.0471, 4.7894, -5.1944],
[ -5.5573, 19.6578, 13.9341]])
In [17]: data.shape
Out[17]: (2, 3)
In [18]: data.dtype
Out[18]: dtype('float64')
1)创建ndarray
np.array(一切对象)—>自动推断数据类型
np.zeros((n,m)) / np.ones((n,m)) / np.empty((n,m)):生成全0,全1,全未初始化垃圾值数组,传入元组/整数
np.arange():range函数的numpy版本 如果没有特别指定,数据类型基本都是float64(浮点数)
2)ndarray的数据类型
dtype是一个特殊的对象:含有ndarray将一块内存解释为特定数据类型所需的信息
—>命名方式:类型名+元素位长(float8字节64位长)
numpy会将 Python类型映射到等价的dtype上(如误写float映射为np.float64)
npo:在此简写为np类型的对象
npo.astype():np强制类型转换
3)Numpy数组的运算
矢量化(vectorization):大小相等的数组之间任何运算均应用到元素级
数组与标量的算术运算会将标量值广播到各个元素
大小相同的数组之间比较会生成布尔数组(逻辑运算)
4)基本的索引和切片
一维数组:
标量赋值切片时,标量值自动广播
数组切片是原始数组的视图,不同于python内置列表,切片上的修改会映射到源数组上
(内置列表的切片是新建一个列表不会改变源列表)——>一种是视图,另一种是副本:副本用npo.copy()
二维数组:
npo[0,2] —> 表示第一行第三列的数
npo[[0,1],2] —> 表示第一二行第三列的数
5)切片索引
一维数组同python列表
二维数组npo[n],n表示行;npo[:n],则表示取数组前n行;npo[:2,1:],表示取前两行,除第一列外的元素数组
6)布尔型索引
一维布尔型数组可用于数组索引:一维布尔型数组长度==索引数组的行长度(轴长度) —>用于筛选
二维则需要大小相同才可做索引运算
比较运算(组合逻辑)得到布尔型索引—>比较运算可以用!= / 也可以用~()对条件进行否定
(用来反转条件好用)
布尔型数组中用 & 和 | ,而不用and or —>数据运算得到的布尔型索引数组是新数组
print(data)
print(data[names == 'tom',1:])
[[ 0.16915065 0.15038684 0.34028728]
[ 0.32732646 0.69241855 0.65203056]
[ 0.84402175 0.77184234 0.48730057]
[ 0.94376757 0.33876278 0.77287885]
[ 0.58535219 0.50321862 0.28196336]]
[[ 0.15038684 0.34028728]
[ 0.50321862 0.28196336]]
(注意: 原书中翻译“通过布尔型索引选取数组中的数据,将总是创建数据的副本,即使返回一模一样的数组也是如此。”<该处翻译可能有误>,因为如下)
In [113]: data[data < 0] = 0
In [114]: data
Out[114]:
array([[ 0.0929, 0.2817, 0.769 , 1.2464],
[ 1.0072, 0. , 0.275 , 0.2289],
[ 1.3529, 0.8864, 0. , 0. ],
[ 1.669 , 0. , 0. , 0.477 ],
[ 3.2489, 0. , 0. , 0.1241],
[ 0.3026, 0.5238, 0.0009, 1.3438],
[ 0. , 0. , 0. , 0. ]])
可以通过布尔索引对数组进行赋值,证明布尔索引得到的数据是视图而非副本
7)花式索引
numpy术语,利用整数数组进行索引
In [119]: arr
Out[119]:
array([[ 0., 0., 0., 0.],
[ 1., 1., 1., 1.],
[ 2., 2., 2., 2.],
[ 3., 3., 3., 3.],
[ 4., 4., 4., 4.],
[ 5., 5., 5., 5.],
[ 6., 6., 6., 6.],
[ 7., 7., 7., 7.]])
In [120]: arr[[4, 3, 0, 6]]
Out[120]:
array([[ 4., 4., 4., 4.],
[ 3., 3., 3., 3.],
[ 0., 0., 0., 0.],
[ 6., 6., 6., 6.]])
花式索引在选择索引行、列上使用频繁(1.一个[]就是一个维度,逗号分割维度 2.按照[]内顺序进行筛选
负数索引从末尾开始选取行
npo[[行],[:,列]],扩展同理(后面格式必须得这样,否则会出现预期不一样的结果
array([[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11],
[12, 13, 14, 15],
[16, 17, 18, 19],
[20, 21, 22, 23],
[24, 25, 26, 27],
[28, 29, 30, 31]])
In [124]: arr[[1, 5, 7, 2], [0, 3, 1, 2]] #这种格式相当于是[1,0],[5,3]....
Out[124]: array([ 4, 23, 29, 10])
In [125]: arr[[1, 5, 7, 2]][:, [0, 3, 1, 2]] #这种格式才是预期的结果,把第一行的第0,3,1,2列都拿出来,再是第五行的....
Out[125]:
array([[ 4, 7, 5, 6],
[20, 23, 21, 22],
[28, 31, 29, 30],
[ 8, 11, 9, 10]])
花式索引!=切片,花式索引是副本而非视图
8)数组转置和轴对换
npo.T属性:数组的转置数组(方法一:针对一二维数组:返回视图非副本
npo.transpose(arry_method)(方法二:针对高维数组:返回视图非副本 arry_method:由轴编号组成的元组 / 轴就是维度
transpose方法结合npo.shape进行 —>
In [60]: arr1
Out[60]: array([[[ 0, 1, 2], [ 3, 4, 5]], [[ 6, 7, 8], [ 9, 10, 11]]])
In [61]: arr1.shape #看形状
Out[61]: (2, 2, 3) #说明这是一个2*2*3的数组(矩阵),返回的是一个元组,可以对元组进行索引,也就是0,1,2,维度进行编号
In [134]: arr.transpose((1, 0, 2)) #这里的(1,0,2)元组就相当是维度元组(0,1,2) 说明把第三维和第二维进行了交换(一般而言保持第一维度不会变化
Out[134]:
array([[[ 0, 1, 2, 3], #比如数值6,原本是(1,0,0)的位置,变换后是(0,1,0)
[ 8, 9, 10, 11]],
[[ 4, 5, 6, 7],
[12, 13, 14, 15]]])
np.dot():计算矩阵内积 —>np.dot(npo.T,npo)
npo.swapaxes(arry_method)(方法三:实现两个维度的交换:返回视图非副本 arry_method:接受一对轴编号 / 就是两个维度
4.2 通用函数:快速的元素级数组函数 ufunc
简单函数(接受一个或多个标量值,并产生一个或多个标量值)的矢量化包装器。
一元ufunc(接受一个矢量):np.sqrt np.exp... 一元:unary
二元ufunc: np.add() np.maximum... 二元:binary
返回多个数组:np.modf(内置函数divmod的矢量化版本:divmod(a,b) / 返回一个包含shan商和余数的元组(a//b,a%b)
//更确切说是math库中modf的矢量化版本:将数组各元素的小数部分和整数部分以两个独立数组形式返回
ufunc可接受一个out可选参数:表示在传入数组上原地进行操作(即结果值返回至数组中
部分一元、二元ufunc
4.3利用数组进行数据处理
np.meshgrid():生成网格点坐标矩阵 //强烈的规律性快速生成矩阵
图中每个交叉点都是网格点;网格点的坐标的矩阵就是坐标矩阵
xs, ys = np.meshgrid(points, points)
传入参数为两个一维数组,分别是所有的x、y值
返回产生的两个二维数组,分别是按网格状化后排列的x、y值
1)将条件逻辑表述为数组运算
np.where():三元表达式 x if condition else y 的矢量化版本
条件:布尔矩阵 if:第二个参数 else:第三个参数 返回:产生的新数组
In [165]: xarr = np.array([1.1, 1.2, 1.3, 1.4, 1.5])
In [166]: yarr = np.array([2.1, 2.2, 2.3, 2.4, 2.5])
In [167]: cond = np.array([True, False, True, True, False]) #cond为True时,选取xarr的值,否则从yarr中选取。
In [168]: result = [(x if c else y) #列表生成式写法
.....: for x, y, c in zip(xarr, yarr, cond)]
In [169]: result
Out[169]: [1.1000000000000001, 2.2000000000000002, 1.3, 1.3999999999999999, 2.5] #1.大数据处理慢 2.无法用于多维数组(多个for时间度问题
In [170]: result = np.where(cond, xarr, yarr) #np.where解决
In [171]: result
Out[171]: array([ 1.1, 2.2, 1.3, 1.4, 2.5])
np.where作用:根据另一个数组而产生一个新数组;
第二三个参数不一定是矢量,也可以是标量
In [173]: arr
Out[173]:
array([[-0.5031, -0.6223, -0.9212, -0.7262],
[ 0.2229, 0.0513, -1.1577, 0.8167],
[ 0.4336, 1.0107, 1.8249, -0.9975],
[ 0.8506, -0.1316, 0.9124, 0.1882]])
In [174]: arr > 0
Out[174]:
array([[False, False, False, False],
[ True, True, False, True],
[ True, True, True, False],
[ True, False, True, True]], dtype=bool)
In [175]: np.where(arr > 0, 2, -2)
Out[175]:
array([[-2, -2, -2, -2],
[ 2, 2, -2, 2],
[ 2, 2, 2, -2],
[ 2, -2, 2, 2]])
2)数学和统计方法
聚合计算:(aggregation,通常叫约简(reduction)) 聚类统计
np.sum np.mean np.std... 求和平均标准差 可以对整个数组 / 也可以接受一个axis计算对某轴的统计值
axis=1(行) axis=0(列)
非聚合计算:np.cumsum np.cumprod... 累加累乘 产生一个结果数组(多维数组中一样,可以指定轴)
3)用于布尔型数组的方法
布尔型数组也可用于统计方法—>转为1 / 0
(sum求True个数
(any是否存在True (all是否都是True 同样适用于数值数组
4)排序
npo.sort():原地排序 (多维数组加axis
np.sort返回副本,npo.sort原地修改本身
计算数组分位数:1.排序 2.选取特定位置值
In [203]: large_arr = np.random.randn(1000)
In [204]: large_arr.sort()
In [205]: large_arr[int(0.05 * len(large_arr))] # 5% quantile
Out[205]: -1.5311513550102103
其他排序:附录+pandas
5)唯一化以及其他的集合逻辑
集合函数
np.unique():找出数组中唯一值并返回排序结果 = set+sorted
np.in1d():试探一个数组的值在另一个数组中的成员是否包含,返回布尔型数组
4.4 用于数组的文件输入输出(较为简单的fIO)
np.save:数组以未压缩的原始二进制格式 扩展名:.npy
np.savez:多个数组保存到一个未压缩文件中,数组以关键字参数形式传入 扩展名:.npz
np.load:加载.npy正常 加载.npz时得到类似字典的对象:对各个数组进行延迟加载(类似迭代器,惰性计算
np.savez_compressed:数据压缩
4.5线性代数
x.dot(y) = np.dot(x,y) :矩阵乘法 所有数组dot方法和np命名空间函数的区别是:dot方法是视图,函数是副本
@符(类似python3.5)也可以作为中缀运算符进行矩阵乘法:
In [230]: x @ np.ones(3)
Out[230]: array([ 6., 15.])
numpy.linalg中有一组标准的矩阵分解运算以及如求逆和行列式等:
In [231]: from numpy.linalg import inv, qr
In [232]: X = np.random.randn(5, 5)
In [233]: mat = X.T.dot(X)
In [234]: inv(mat)
Out[234]:
array([[ 933.1189, 871.8258, -1417.6902, -1460.4005, 1782.1391],
[ 871.8258, 815.3929, -1325.9965, -1365.9242, 1666.9347],
[-1417.6902, -1325.9965, 2158.4424, 2222.0191, -2711.6822],
[-1460.4005, -1365.9242, 2222.0191, 2289.0575, -2793.422 ],
[ 1782.1391, 1666.9347, -2711.6822, -2793.422 , 3409.5128]])
In [235]: mat.dot(inv(mat)) #求逆
Out[235]:
array([[ 1., 0., -0., -0., -0.],
[-0., 1., 0., 0., 0.],
[ 0., 0., 1., 0., 0.],
[-0., 0., 0., 1., -0.],
[-0., 0., 0., 0., 1.]])
In [236]: q, r = qr(mat) #qr矩阵分解
In [237]: r
Out[237]:
array([[-1.6914, 4.38 , 0.1757, 0.4075, -0.7838],
[ 0. , -2.6436, 0.1939, -3.072 , -1.0702],
[ 0. , 0. , -0.8138, 1.5414, 0.6155],
[ 0. , 0. , 0. , -2.6445, -2.1669],
[ 0. , 0. , 0. , 0. , 0.0002]])
```![](https://img2018.cnblogs.com/blog/1816059/201910/1816059-20191006224358687-830915484.png)
##4.6 伪随机数生成
np.random.normal:标准正态分布
np.random.seed:设置随机数生成种子(全局状态种子
np.random.RandomState:创建一个与其他隔离的随机数生成器(局部状态种子)
![](https://img2018.cnblogs.com/blog/1816059/201910/1816059-20191006224421566-100182321.png)
![](https://img2018.cnblogs.com/blog/1816059/201910/1816059-20191006224427232-938067935.png)
##4.7 示例:随机漫步
纯python:
In [247]: import random
.....: position = 0
.....: walk = [position]
.....: steps = 1000
.....: for i in range(steps):
.....: step = 1 if random.randint(0, 1) else -1
.....: position += step
.....: walk.append(position)
.....:
转为numpy:
In [251]: nsteps = 1000
In [252]: draws = np.random.randint(0, 2, size=nsteps) #size决定是一维的
In [253]: steps = np.where(draws > 0, 1, -1)
In [254]: walk = steps.cumsum()
In [249]: plt.plot(walk[:100]) #x步数也就是时间,y是离初始点的距离
![](https://img2018.cnblogs.com/blog/1816059/201910/1816059-20191006224440518-1952102675.png)
对l累加后的漫步数据一些统计计算处理:
In [255]: walk.min()
Out[255]: -3
In [256]: walk.max()
Out[256]: 31
假设离初始点距离10步外就是穿越过了漫步地点,想要知道首次穿越时间:
In [257]: (np.abs(walk) >= 10).argmax() #argmax返回数组最大值的第一个索引位置
Out[257]: 37 #所以第37步时第一次走出漫步地点
一次性模拟多个随机漫步:
In [258]: nwalks = 5000
In [259]: nsteps = 1000
In [260]: draws = np.random.randint(0, 2, size=(nwalks, nsteps)) # 0 or 1 #size决定是二维的 行:第几次walk 列:nwalk的第几步
In [261]: steps = np.where(draws > 0, 1, -1)
In [262]: walks = steps.cumsum(1) #按行作累加 / 非聚合
In [263]: walks
Out[263]:
array([[ 1, 0, 1, ..., 8, 7, 8],
[ 1, 0, -1, ..., 34, 33, 32],
[ 1, 0, -1, ..., 4, 5, 4],
...,
[ 1, 2, 1, ..., 24, 25, 26],
[ 1, 2, 3, ..., 14, 13, 14],
[ -1, -2, -3, ..., -24, -23, -22]])
计算所有随机漫步的统计数据:
In [264]: walks.max()
Out[264]: 138
In [265]: walks.min()
Out[265]: -133
之后计算每个随机漫步的首次穿越时间:阈值=30(注:不是每次漫步都能够穿越,所以需要清理检查
In [266]: hits30 = (np.abs(walks) >= 30).any(1) #按行筛选每行是不是有abs>30的值
In [267]: hits30
Out[267]: array([False, True, False, ..., False, True, False], dtype=bool)
In [268]: hits30.sum() # Number that hit 30 or -30 #为True的就是满足要求的,一共有3410个
Out[268]: 3410
获取穿越时间:
In [269]: crossing_times = (np.abs(walks[hits30]) >= 30).argmax(1) #花式索引筛选满足要求行+argmax找到首次穿越的时间
In [270]: crossing_times.mean()
Out[270]: 498.88973607038122