Python 数据分析 NumPy

https://www.runoob.com/numpy

NumPy是Python的一种开源的数值计算扩展。这种工具可用来存储和处理大型矩阵,比Python自身的嵌套列表(nested list structure)结构要高效的多。

NumPy(Numeric Python)提供了许多高级的数值编程工具。Numpy的一个重要特征是它的数组计算,是我们做数据分析必不可少的一个包。

导入Python库使用关键字import,后面可以自定义库的简称,但是一般都将NumPy命名为np,pandas命名为pd。

使用前一定要先导入Numpy包,导入的方法有以下几种:

import numpy

import numpy as np

from numpy import * ##不用加前缀,但和其他包函数重名容易报错

import numpy as np

1Numpy对象及其索引

数组上的数学操作

假如我们想将列表中的每个元素增加1,但列表不支持这样的操作:

a=[1,2,3,4]
[i+1 for i in a]
[2, 3, 4, 5]

与另一个数相加,得到对应元素相加的结果:

a=[1,2,3,4]
b=[2,3,4,5]
[x+y for (x,y) in zip(a,b)]
[3, 5, 7, 9]

zip() 函数用于将可迭代的对象作为参数,将对象中对应的元素打包成一个个元组,然后返回由这些元组组成的列表。

这样的操作比较麻烦,而且在数据量特别大的时候会非常耗时间。

如果我们使用Numpy,就会变得特别简单

a=np.array([1,2,3,4])
b=np.array([2,3,4,5])
a+1
array([2, 3, 4, 5])
a+b
array([3, 5, 7, 9])

产生数组 array

1.从列表产生数据

list=[0,1,2,3]
a=np.array(list)
a
array([0, 1, 2, 3])

2.从列表传入

a=np.array([1,2,3,4])
a
array([1, 2, 3, 4])

3.生成全0数组

a=np.zeros(5)
a
array([0., 0., 0., 0., 0.])

4.生成全1数组

a=np.ones(5,dtype='bool')   ##dtype定义需要的数据类型 这里定义成布尔
a
array([ True,  True,  True,  True,  True])

5.使用fill方法将数组设定为指定值

与列表不同,数组中要求所有元素的dtype是一样的,如果传入参数的类型与数组类型不一样,需要按照已有的类型进行转换。

a=np.array([1,2,3,4])
a.fill(2)
a
array([2, 2, 2, 2])
a.fill(2.5)   ##传入的参数类型会按照原数组类型进行转换 浮点型转整型
a
array([2, 2, 2, 2])
##使用fill方法将原数组类型改变
a=np.array([1,2,3,4])
a=a.astype('float')
a.fill(2.5)
a
array([2.5, 2.5, 2.5, 2.5])

使用特定的方法生成特殊的数组

1.生成整数序列 arange

a=np.arange(1,10)  
a
array([1, 2, 3, 4, 5, 6, 7, 8, 9])
a=np.arange(1,10,2)  ##arange(起点,终点,步长)
a
array([1, 3, 5, 7, 9])

2.生成等差数列 linspace

a=np.linspace(1,19,10)   ##linspace(起点,终点,个数),此处从1开始到19结束,生成10个元素
a
array([ 1.,  3.,  5.,  7.,  9., 11., 13., 15., 17., 19.])

3.生成随机数 random.rand/randn

a=np.random.rand(10)  ##生成0-1的10个随机数 浮点数
a
array([0.06818817, 0.33335218, 0.64765711, 0.4336125 , 0.46251429,
       0.78930739, 0.47116701, 0.60252559, 0.77941471, 0.57273979])
b=np.random.randn(10)  ##生成0-1的10个随机数,随机数服从标准正太分布
b
array([ 0.5770561 , -1.36047333, -0.41379915, -0.4660371 , -0.24865449,
        0.99519499, -1.27109413, -1.18072939, -0.75014523,  0.17017929])
c=np.random.randint(1,10,10)  ##生成1-10的10个随机整数 randint(起点,终点,个数)
c
array([6, 2, 4, 3, 9, 6, 3, 9, 7, 4])

数组属性

a
array([0.06818817, 0.33335218, 0.64765711, 0.4336125 , 0.46251429,
       0.78930739, 0.47116701, 0.60252559, 0.77941471, 0.57273979])

1.查看类型

a=np.array([1,2,3,4])
type(a)
numpy.ndarray

numpy的n维数组

2.查看数组中的数据类型

a.dtype
dtype('int32')

3.查看形状,会返回一个元组,每个元素代表这一堆的元素数目

a.shape
(4,)

查看数组里面元素的数目

a.size
4

查看数组的维度

a.ndim
1

索引与切片

索引第一个元素

a=np.array([1,2,3,4])
a[0]
1

修改第一个元素的值

a[0]=10
a
array([10,  2,  3,  4])

切片,支持负索引

a=np.array([11,12,13,14,15])
a[1:3]
array([12, 13])
a[1:-2] # 第一个元素 到 倒数第二个元素的前一个元素
array([12, 13])

省略参数

a=np.array([11,12,13,14,15])
a[-2:]
array([14, 15])
a[::2]
array([11, 13, 15])

错位相减
假设我们记录一部电影的累计票房:

ob=np.array([21000,21800,22100,23400,25000])
ob
array([21000, 21800, 22100, 23400, 25000])

可以这样计算每天的票房

ob2=ob[1:]-ob[:-1]
ob2
array([ 800,  300, 1300, 1600])

多维数组

多维数组及其属性

将以列表为元素的列表传入,最终得到的额就是一个二维数组

a=np.array([[1,2,3,4],[5,6,7,8]])
a
array([[1, 2, 3, 4],
       [5, 6, 7, 8]])

查看形状

a.shape #两行四列 2*4
(2, 4)

查看总的元素个数

a.size 
8

查看维数

a.ndim
2
多维数组索引

对于二维数组,可以传入两个数字来索引

a=np.array([[1,2,3,4],[5,6,7,8]])
a
array([[1, 2, 3, 4],
       [5, 6, 7, 8]])
a[1,3]
8

索引赋值

a[1,3]=9
a
array([[1, 2, 3, 4],
       [5, 6, 7, 9]])

索引一整行内容

a[1]
array([5, 6, 7, 8])

Python会将这单个元组当成对第一堆的索引,然后返回对应的内容

行索引在逗号前,列索引在逗号后

a[:,1]  ##返回的是第二列。
array([2, 6])
多维数组切片
a=np.array([[0,1,2,3,4,5],[10,11,12,13,14,15],[20,21,22,23,24,25],[30,31,32,33,34,35],[40,41,42,43,44,45],[50,51,52,53,54,55]])
a
array([[ 0,  1,  2,  3,  4,  5],
       [10, 11, 12, 13, 14, 15],
       [20, 21, 22, 23, 24, 25],
       [30, 31, 32, 33, 34, 35],
       [40, 41, 42, 43, 44, 45],
       [50, 51, 52, 53, 54, 55]])

取到第一行的第4和第5两个元素
:左闭右开的区间

a[0,3:5]
array([3, 4])

得到最后两行的最后两列

a[-2:,-2:]
array([[44, 45],
       [54, 55]])

得到第3列

a[:,2]
array([ 2, 12, 22, 32, 42, 52])

每一堆都支持切片的规则,包括负索引,省略

[起点:终点:步长]左闭右开
#例如:取出3,5行的奇数列
a[2:5:2,::2]
array([[20, 22, 24],
       [40, 42, 44]])

切片是引用

切片在内存中使用的是引用机制。

引用机制意味着,Python并没有为b分配新的空间来存储它的值,而实让b指向了a所分配的内存空间,因此,改变b会改变a的值。

a=np.array([0,1,2,3,4])
b=a[2:4]
b
array([2, 3])
b[0]=10
a
array([ 0,  1, 10,  3,  4])

而这种现象在列表中并不会出现

list=[1,2,3,4,5]
list1=list[2:4]
list1
[3, 4]
list1[0]=10
list
[1, 2, 3, 4, 5]

这样的好处在于,对于很大的数据,不用大量复制多余的值,节约了空间

缺点在于,可能出现改变一个值而改变另一个值的情况。

一个解决方法是使用copy()方法产生一个复制,这个复制中会申请新的内存。

a=np.array([0,1,2,3,4])
b=a[2:4].copy()
b[0]=10
a
array([0, 1, 2, 3, 4])

花式索引

切片只能支撑连续或者等间隔的切片操作,要想实现任意位置的操作,需要使用花式索引 fancy slicing

一维花式索引

与range函数类似,我们可以使用arange函数来产生等差数组

a=np.arange(0,100,10)
a
array([ 0, 10, 20, 30, 40, 50, 60, 70, 80, 90])

花式索引需要指定索引位置

index=[1,2,-3]
y=a[index]
y
array([10, 20, 70])

还可以使用布尔数组来花式索引

mask=np.array([0,2,2,0,0,1,0,0,1,0],dtype=bool)
mask
array([False,  True,  True, False, False,  True, False, False,  True,
       False])
y=a[mask]
y
array([10, 20, 50, 80])

mask必须是布尔数组,长度必须和数组长度相等

二维花式索引

对于二维花式索引,我们需要给定行和列的值

a=np.array([[0,1,2,3,4,5],[10,11,12,13,14,15],[20,21,22,23,24,25],[30,31,32,33,34,35],[40,41,42,43,44,45],[50,51,52,53,54,55]])
a
array([[ 0,  1,  2,  3,  4,  5],
       [10, 11, 12, 13, 14, 15],
       [20, 21, 22, 23, 24, 25],
       [30, 31, 32, 33, 34, 35],
       [40, 41, 42, 43, 44, 45],
       [50, 51, 52, 53, 54, 55]])

返回的是一条次对角线上的5个值(此处取1,12,23,34,45)

a[(0,1,2,3,4),(1,2,3,4,5)]   ##两个括号相对位置的数分别为行标和列标
array([ 1, 12, 23, 34, 45])

返回最后三行的1,3,5列

a[-3:,0:6:2]
array([[30, 32, 34],
       [40, 42, 44],
       [50, 52, 54]])
a[-3:,[0,2,4]]
array([[30, 32, 34],
       [40, 42, 44],
       [50, 52, 54]])

使用布尔数组进行索引

mask=np.array([1,0,1,0,1,0],dtype=bool)
a[-3:,mask]
array([[30, 32, 34],
       [40, 42, 44],
       [50, 52, 54]])

与切片不同,花式索引返回的是对象的一个复制而不是引用

“不完全”索引

只给定行索引的时候,返回整行

a=np.array([[0,1,2,3,4,5],[10,11,12,13,14,15],[20,21,22,23,24,25],[30,31,32,33,34,35],[40,41,42,43,44,45],[50,51,52,53,54,55]])
y=a[:3]
y
array([[ 0,  1,  2,  3,  4,  5],
       [10, 11, 12, 13, 14, 15],
       [20, 21, 22, 23, 24, 25]])

这时候也可以使用花式索引取出2,3,5行

con=np.array([0,1,1,0,1,0],dtype=bool)
a[con]
array([[10, 11, 12, 13, 14, 15],
       [20, 21, 22, 23, 24, 25],
       [40, 41, 42, 43, 44, 45]])

where语句

where(array)
where函数会返回所有非零元素的索引
满足条件的位置的元组

a=np.random.randint(1,10,10)
a
array([2, 1, 8, 9, 2, 3, 7, 7, 4, 3])
np.where(a>5)
(array([2, 3, 6, 7], dtype=int64),)
np.where(a>5,1,0)
array([0, 0, 1, 1, 0, 0, 1, 1, 0, 0])

np.where(condition[,x,y])
三个参数满足条件condition时输出x,不满足时输出y
这里元素大于5 满足输出1 否则输出0

一维数组
a=np.array([0,12,5,20])

判断句数组中的元素是不是大于10

a>10
array([False,  True, False,  True])

数组中所有大于10的元素的索引位置

np.where(a>10)
(array([1, 3], dtype=int64),)

注意到where的返回值是一个元组,返回的是索引位置

也可以直接用数组操作

a[a>10]
array([12, 20])
a[np.where(a>10)]
array([12, 20])

2数组类型

数组类型

布尔型:bool:占一个字节

整型:int8/16/32/64/128,int:和C语言的long一样大

无符号整型:uint8/16/32/64/128,uint:和C语言中的unsigned long一样大

浮点数:float16/32/64,float,longfloat:默认为双精度float64,longfloat精度大小与系统有关

复数:complex64/128,complex,longconplex:默认为complex128,即实部虚部都为双精度

字符串:string,unicode:可以使用表示一个4字节字符串的数组

对象:object:数组中可以使用任意值

时间:datetime64,timedelta64

类型转换

a=np.array([1.5,-3],dtype=int)
a
array([ 1, -3])

asarray函数
numpy.asarray(a, dtype=None, order=None)
将输入转换为数组。

a=np.array([1,2,3])
np.asarray(a,dtype=float)
array([1., 2., 3.])

astype函数

a=np.array([1,2,3])
a.astype(float)
array([1., 2., 3.])
##astype方法返回一个新数组 而不是引用
a=np.array([1,2,3])
b=a.astype(float)
b[0]=0.5
b
array([0.5, 2. , 3. ])
a
array([1, 2, 3])

astype引用不会对原数组产生影响

3 数组操作

以豆瓣10部高分电影为例

##电影名称
mv_name=['肖申克的救赎','控方证人','美丽人生','阿甘正传','霸王别姬','泰坦尼克号','辛德勒的名单','这个杀手不太冷','疯狂动物成','海豚湾']
##评分人数
mv_num=np.array([692795,42995,327855,580897,478523,157074,306904,662552,284652,159302])
##评分
mv_score=np.array([9.6,9.5,9.5,9.4,9.4,9.4,9.4,9.3,9.3,9.3])
##电影时长
mv_length=np.array([142,116,116,142,171,194,195,133,109,92])

数组排序

sort函数

##对评分人数排序
np.sort(mv_num)
array([ 42995, 157074, 159302, 284652, 306904, 327855, 478523, 580897,
       662552, 692795])
mv_num  ##sort函数不改变原数组
array([692795,  42995, 327855, 580897, 478523, 157074, 306904, 662552,
       284652, 159302])

argsort返回从小到大的排列在数组中的索引位置

argsort函数

order=np.argsort(mv_num)
order
array([1, 5, 9, 8, 6, 2, 4, 3, 7, 0], dtype=int64)
##看评分人数最少
order[0]
1
##通过order获得索引,然后查看对应索引的其他数组信息
mv_name[1]
'控方证人'

数学操作

求和

np.sum(mv_num)
3693549
mv_num.sum()
3693549

最大值

mv_length.max()
195
np.max(mv_length)
195

最小值 min

均值 mean

标准差 std

方差 var

相关系数矩阵 cov

np.cov(mv_score,mv_length)
array([[9.88888889e-03, 4.55555556e-01],
       [4.55555556e-01, 1.26288889e+03]])

多维数组操作

数组形状 shape
a=np.arange(6)
a
array([0, 1, 2, 3, 4, 5])
##修改原数组成2行3列
a.shape=2,3
a
array([[0, 1, 2],
       [3, 4, 5]])
a.shape
(2, 3)

与之对应的方法是reshape,但它不会改变原来数组的值,而是返回一个新的数组

a=np.arange(6)
a
array([0, 1, 2, 3, 4, 5])
a.reshape(2,3)
array([[0, 1, 2],
       [3, 4, 5]])
a
array([0, 1, 2, 3, 4, 5])
转置 T/transpose
a=a.reshape(2,3)
a
array([[0, 1, 2],
       [3, 4, 5]])
a.T
array([[0, 3],
       [1, 4],
       [2, 5]])
a.transpose()
array([[0, 3],
       [1, 4],
       [2, 5]])

4数组连接

concatenate((a0,a1,...,an),axis=0)

axis=0代表往跨行(down),而axis=1代表跨列(across)

使用0值表示沿着每一列或行标签\索引值向下执行方法

使用1值表示沿着每一行或者列标签模向执行对应的方法

注意:这些数组要用()包括到一个元组中去。

除了给定的轴外,这些数组其他轴的长度必须是一样的。

x=np.array([[0,1,2],[10,11,12]])
y=np.array([[50,51,51],[60,61,62]])
print(x.shape,y.shape)
(2, 3) (2, 3)

默认沿着第一维进行连接

z=np.concatenate((x,y),axis=0)
z
array([[ 0,  1,  2],
       [10, 11, 12],
       [50, 51, 51],
       [60, 61, 62]])

沿着第二维进行连接

z=np.concatenate((x,y),axis=1)
z
array([[ 0,  1,  2, 50, 51, 51],
       [10, 11, 12, 60, 61, 62]])

注意到这里x和y的形状是一样的,还可以将它们连接成三维的数组,但是concatenate不提供这样的功能,可以这样操作:

z=np.array((x,y))
z
array([[[ 0,  1,  2],
        [10, 11, 12]],

       [[50, 51, 51],
        [60, 61, 62]]])

事实上,NumPy提供了分别对应这三种情况的函数

vstack

np.vstack((x,y))
array([[ 0,  1,  2],
       [10, 11, 12],
       [50, 51, 51],
       [60, 61, 62]])

hstack

np.hstack((x,y))
array([[ 0,  1,  2, 50, 51, 51],
       [10, 11, 12, 60, 61, 62]])

dstack

np.dstack((x,y))
array([[[ 0, 50],
        [ 1, 51],
        [ 2, 51]],

       [[10, 60],
        [11, 61],
        [12, 62]]])

5 NumPy内置函数

https://blog.csdn.net/nihaoxiaocui/article/details/51992860

a=np.array([-1,2,3,-2])
##绝对值
np.abs(a)
array([1, 2, 3, 2])
##指数
np.exp(a)
array([ 0.36787944,  7.3890561 , 20.08553692,  0.13533528])
##中值
np.median(a)
0.5
##累计和(从前到后依次累加)
np.cumsum(a)
array([-1,  1,  4,  2], dtype=int32)
posted @ 2021-06-01 16:19  王陸  阅读(347)  评论(0编辑  收藏  举报