Numpy, 数组和矢量计算包
前几年前想学数据分析,于是就去学习Python的Numpy。然而看完《利用Python进行数据分析》后,也对它没有多大印象的。但是学了一段时间R语言,并且将R语言和Python进行一些比较,再次学习Numpy就特别轻松了。
由于学过R语言,我可以简单认为Numpy提供的多维数据对象ndarray就是Python版本的R语言的vector
, matrix
和array
。几乎没有特殊说明,这两者的任何操作都是一致的。
Numpy的部分功能如下:
- ndarray, 一个具有矢量算术运算和复杂广播能力的快速且节省空间的多维数组
- 用于对整组数据进行快速运算的标准数学函数(无需编写循环函数)
- 用于读写磁盘数据的工具以及用于操作内存映射文件的工具
- 线性代数, 随机数生成以及傅里叶变换功能
- 用于继承由C, C++, Fortran等语言编写的代码的工具
约定俗成:
import numpy as np
也就说不会特意去声明导入numpy
。
创建多维数组
多维数组有多种创建方法,其中最简单的就是使用arrary
函数。以一切序列类型的对象作为输入
# 一维数组,也就是R语言的最基本元素,vector
In [1]: import numpy as np
In [2]: data1 = [1,2,3,4,5]
In [3]: arr1 = np.array(data1)
In [4]: ?arr1 # 内省下arr1对象
# 二维数组,也就是R语言的matrix
In [5]: data2 = [[1,2,3],[4,5,6]]
In [6]: arr2 = np.array(data2)
In [7]: ?arr2
# 在R语言中用dim(), nrows, ncols查看数据维度
# 在Python中,这些可以用对象的方法查看。
In [10]: arr2.ndim
Out[10]: 2
In [11]: arr2.shape
Out[11]: (2, 3)
# 和R语言一样,ndarray的对象不允许存在多种数据类型
# ndarray会自动根据输入选择最合适的数据类型
In [17]: data3 = [1,"string", True]
In [18]: arr3 = np.array(data3)
In [19]: arr3
Out[19]:
array(['1', 'string', 'True'],
dtype='<U11')
In [20]: arr3.dtype
Out[20]: dtype('<U11')
除了用array
转换序列型数据输入以外,还可以用arange
(类似于内置的ranges), asarray
(类似于array)。后面的方法和线性代数密切相关,建议查看相应的说明: ones ones_like
, zeros zeros_like
empty empty_like
,eye identity
。
关于数据类型, 一般情况下我们没必要对它太过于关注。但是对于大数据集,则需要自己主动声明。因为数据类型(dtype)负责将一块内存解释为特定数据类型,即直接映射到相应的机器表示。在R语言中有一类类型转换函数(例如as.numeric)对数组内的数据类型进行转换,在Numpy则通过dtype.
数组运算
R语言的一大特点就是矢量化运算,能用来检查你是否理解R语言。简单理解,就是不用循环就能对数据批量运算。
个人愚见:矢量化运算是Numpy用C语言编写,在C语言层面是也是循环。这也是为什么一个数组内的数据类型要一致。
# R
> arr1
[1] 1 2 3 4 5
> arr1 + 1
[1] 2 3 4 5 6
# Python
In [29]: arr1 + 1
Out[29]: array([2, 3, 4, 5, 6])
In [35]: arr2 * 3
Out[35]: array([[ 3, 6, 9],
[12, 15, 18]])
In [36]: arr2 * arr2
Out[36]: array([[ 1, 4, 9],
[16, 25, 36]])
我曾经在 Python和R的异同(一)里谈到原生Python要想实现R语言的矢量化就要使用列表推导式, 而目前可以用numpy带来的矢量化运算属性了。
索引和切片
切片
在R语言和Numpy,包括原生的Python都有切片的功能, 所谓的切片(slicing) 就是从已有的数组中返回选定的元素,而索引(index)提供指向存储在数组指定位置的数据值的指针。
# R
arr <- 0:9
arr_sub <- arr[1:5]
# Python default list
data = [i for i in range(10)]
data_sub = data[0:5]
# Python Numpy
arr = np.arange(10)
arr_sub = arr[0:5]
上面的结果都是一致的,都是提取前5个元素。只不过要注意R语言的索引从1开始(5-1+1),而Python从0开始(5-0)。表面看起来是相同的,但其实Numpy切片得到只是原始数据的视图(view),也就是浅复制,即你对Numpy切片后的数据进行操作,会影响到原始数据。
# Python Numpy
In [60]: arr_sub[1] = 100
In [61]: arr
Out[61]: array([ 0, 100, 2, 3, 4, 5, 6, 7, 8, 9])
# Python default list
In [65]: data_sub[1] = 100
In [66]: data_sub
Out[66]: [0, 100, 2, 3, 4]
In [67]: data
Out[67]: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
# R 同Python默认的列表
原因就是Numpy的目的是处理大数据,对大规模的数据进行实际复制会消耗不必要的性能和内存。
numpy的索引操作和R语言几乎一模一样,分为切片索引,布尔值索引,花式索引。这些都在《R语言实战》基本数据管理章节中的数据集选取子集里面提及。
切片索引:
In [79]: arr = np.eye(9,9)
## 类似于R的操作
In [80]: arr[1,1]
Out[80]: 1.0
In [82]: arr[:,:]
## Python原来是通过递归对元素进行访问
In [81]: arr[1][1]
Out[81]: 1.0
布尔值索引, 也就是先产生一个True, False的数组,然后根据这个数组提取数据
In [87]: arr[arr == 1]
Out[87]: array([ 1., 1., 1., 1., 1., 1., 1., 1., 1.])
花式索引, 就是提供指定顺序的整数型列表
In [94]: arr = np.empty((8,4))
In [95]: for i in range(8):
...: arr[i] = i
...:
## 选取第4,3,1,6行数据
In [99]: arr[[3,2,0,5]]
Out[99]: array([ 3., 2., 0., 5.])
## 在R里面就是
## arr[c(4,3,1,6)]
注意,如果一次性传入多个索引数据,Numpy会返回一维数组,但是R依旧会返回多维。这是目前第一个与R不太一样,当然和预想的结果也不同。
# R
> mdata <- matrix(1:32,nrow=8, ncol=4)
> mdata[c(2,6,8,2),c(1,4,2,3)]
[,1] [,2] [,3] [,4]
[1,] 2 26 10 18
[2,] 6 30 14 22
[3,] 8 32 16 24
[4,] 2 26 10 18
# Numpy
In [110]: arr = np.arange(1,33).reshape(8,4)
In [111]: arr[[1,5,7,2],[0,3,1,2]]
Out[111]: array([ 5, 24, 30, 11])
## 为了解决这个问题,有两种方法
In [112]: arr[[1,5,7,2]][:,[0,3,1,2]]
Out[112]:
array([[ 5, 8, 6, 7],
[21, 24, 22, 23],
[29, 32, 30, 31],
[ 9, 12, 10, 11]])
In [113]: arr[np.ix_([1,5,7,2],[0,3,1,2])]
Out[113]:
array([[ 5, 8, 6, 7],
[21, 24, 22, 23],
[29, 32, 30, 31],
[ 9, 12, 10, 11]])
注: 花式索引以及布尔值索引和切片索引不同, 前者将数据复制到新的数组中,而后者是原始数据的视图。 可能原因是前两者的得到数据在原始数据中位置不是整块存放。
数据转置和轴对换
转置(transpose)是数据重塑的一种特殊形式,返回的是原始数据的视图(这一点和R不同)。数组不仅有transpose方法,还有一个T属性, 这两者在二维数组上是相同的。
arr = np.arange(1,33).reshape(8,4)
arr.T
np.transpose(arr)
# 线性代数的矩阵内积
np.dot(arr.T, arr)
但是在更高维度上,T属性依旧还是轴对换,transpose方法还需要提供轴编号组成的元组,这个真的是非常难以理解。
通用函数:快速的元素级数组函数
我曾经写过一篇文章,叫做R语言的数据管理里面提到了基石函数,来源于《R语言实战》的高级数学管理。在numpy,这类函数叫做通用函数(UNIVERSAL FUNCTIONS, UFUNC),能够进行矢量化运算的函数。按照官方文档的划分,大致分为
- 数学运算
- 三角函数
- 位运算函数
- 比较函数
- 浮点函数
按照《利用Pyton进行数据分析》可以分为一元函数和二元函数。
对于一些自定义的函数,R语言采用的apply家族函数进行矢量化操作,避免循环。而在Numpy则是frompyfunc
。不过这已经比较高级了。
Numpy更多是Python进行科学计算的基础包,因此数据分析部分的内容就交给pandas吧。