Numpy——快速处理数据

怼烎@ 2022-05-19 13:43 242阅读 0赞

###### NumPy提供了两种基本的对象：ndarray（N-dimensional array object）和 ufunc（universal function object）。ndarray是存储单一数据类型的多维数组，而ufunc则是能够对数组进行处理的函数。 ######

## 1.ndrray对象 ##

函数库的导入

import numpy as np

#### （1）创建 ####

首先需要创建数组才能对其进行其他操作。  
我们可以通过给array函数传递Python的序列对象创建数组，如果传递的是多层嵌套的序列，将创建多维数组(下例中的变量c)。  
![这里写图片描述][70]  
数组的大小可以通过其shape属性获得:  
![这里写图片描述][70 1]

#### （2）存取元素 ####

>>> a = np.arange(10)
    >>> a[5] # 用整数作为下标可以获取数组中的某个元素
    5
    >>> a[3:5] # 用范围作为下标获取数组的一个切片，包括a[3]不包括a[5]
    array([3, 4])
    >>> a[:5] # 省略开始下标，表示从a[0]开始
    array([0, 1, 2, 3, 4])
    >>> a[:-1] # 下标可以使用负数，表示从数组后往前数
    array([0, 1, 2, 3, 4, 5, 6, 7, 8])
    >>> a[2:4] = 100,101 # 下标还可以用来修改元素的值
    >>> a
    array([ 0, 1, 100, 101, 4, 5, 6, 7, 8, 9])
    >>> a[1:-1:2] # 范围中的第三个参数表示步长，2表示隔一个元素取一个元素
    array([ 1, 101, 5, 7])
    >>> a[::-1] # 省略范围的开始下标和结束下标，步长为-1，整个数组头尾颠倒
    array([ 9, 8, 7, 6, 5, 4, 101, 100, 1, 0])
    >>> a[5:1:-2] # 步长为负数时，开始下标必须大于结束下标
    array([ 5, 101])

\*\*使用整数序列 \*\*  
当使用整数序列对数组元素进行存取时，将使用整数序列中的每个元素作为下标，整数序列可以是列表或者数组。使用整数序列作为下标获得的数组不和原始数组共享数据空间。

>>> x = np.arange(10,1,-1)
    >>> x
    array([10, 9, 8, 7, 6, 5, 4, 3, 2])
    >>> x[[3, 3, 1, 8]] # 获取x中的下标为3, 3, 1, 8的4个元素，组成一个新的数组
    array([7, 7, 9, 2])
    >>> b = x[np.array([3,3,-3,8])] #下标可以是负数
    >>> b[2] = 100
    >>> b
    array([7, 7, 100, 2])
    >>> x # 由于b和x不共享数据空间，因此x中的值并没有改变
    array([10, 9, 8, 7, 6, 5, 4, 3, 2])
    >>> x[[3,5,1]] = -1, -2, -3 # 整数序列下标也可以用来修改元素的值
    >>> x
    array([10, -3, 8, -1, 6, -2, 4, 3, 2])

**使用布尔序列**  
当使用布尔数组b作为下标存取数组x中的元素时，将收集数组x中所有在数组b中对应下标为True的  
元素。使用布尔数组作为下标获得的数组不和原始数组共享数据空间，注意这种方式只对应于布尔数组，不能使用布尔列表。

>>> x = np.arange(5,0,-1)
    >>> x
    array([5, 4, 3, 2, 1])
    >>> x[np.array([True, False, True, False, False])]
    >>> # 布尔数组中下标为0，2的元素为True，因此获取x中下标为0,2的元素
    array([5, 3])
    >>> x[[True, False, True, False, False]]
    >>> # 如果是布尔列表，则把True当作1, False当作0，按照整数序列方式获取x中的元素
    array([4, 5, 4, 5, 5])
    >>> x[np.array([True, False, True, True])]
    >>> # 布尔数组的长度不够时，不够的部分都当作False
    array([5, 3, 2])
    >>> x[np.array([True, False, True, True])] = -1, -2, -3
    >>> # 布尔数组下标也可以用来修改元素
    >>> x
    array([-1, 4, -2, -3, 1])

#### （3）多维数组 ####

多维数组的存取和一维数组类似，因为多维数组有多个轴，因此它的下标需要用多个值来表示，NumPy采用组元(tuple)作为数组的下标。如下图所示，a为一个6x6的数组，图中用颜色区分了各个下标以及其对应的选择区域。  
![这里写图片描述][70 2]

#### （4）结构数组 ####

在C语言中我们可以通过struct关键字定义结构类型，结构中的字段占据连续的内存空间，每个结构体占用的内存大小都相同，因此可以很容易地定义结构数组。和C语言一样，在NumPy中也很容易对这种结构数组进行操作。只要NumPy中的结构定义和C语言中的定义相同，NumPy就可以很方便地读取C语言的结构数组的二进制数据，转换为NumPy的结构数组。  
\#\#\#\#（5）内存结构  
如下图所示，这个结构引用了两个对象：一块用于保存数据的存储区域和一个用于描述元素类型的dtype对象。  
![这里写图片描述][70 3]  
图中显示的是如下数组的内存结构：

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

## 2.ufunc运算 ##

ufunc是universal function的缩写，它是一种能对数组的每个元素进行操作的函数。  
**数组对象支持的操作符：**

<table> 
 <thead> 
  <tr> 
   <th>y=x1+x2 :</th> 
   <th>add(x1,x2 [,y])</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <td>y=x1-x2 :</td> 
   <td>subtract(x1,x2 [,y])</td> 
  </tr> 
  <tr> 
   <td>y=x1*x2 :</td> 
   <td>multiply(x1,x2 [,y])</td> 
  </tr> 
  <tr> 
   <td>y=x1/x2 :</td> 
   <td>divide(x1,x2 [,y]),如果两个数的元素为整数，那么用整数除法</td> 
  </tr> 
  <tr> 
   <td>y=x1/x2 :</td> 
   <td>true divide(x1,x2 [,y])，总是返回精确的商</td> 
  </tr> 
  <tr> 
   <td>y=x1//x2 :</td> 
   <td>floor divide(x1,x2 [,y])，总是对返回值取整</td> 
  </tr> 
  <tr> 
   <td>y=-x :</td> 
   <td>negative(x[,y])</td> 
  </tr> 
  <tr> 
   <td>y=x1**x2 :</td> 
   <td>power(x1,x2 [,y])</td> 
  </tr> 
  <tr> 
   <td>y=x1%x2 :</td> 
   <td>remander(x1,x2 [,y]),mod(x1,x2 [,y])</td> 
  </tr> 
 </tbody> 
</table>

#### （1）广播 ####

当我们使用ufunc函数对两个数组进行计算时，ufunc函数会对这两个数组的对应元素进行计算，因此它要求这两个数组有相同的大小(shape相同)。如果两个数组的shape不同的话，会进行如下的广播(broadcasting)处理：  
① 让所有输入数组都向其中shape最长的数组看齐，shape中不足的部分都通过在前面加1补齐  
②输出数组的shape是输入数组shape的各个轴上的最大值  
③如果输入数组的某个轴和输出数组的对应轴的长度相同或者其长度为1时，这个数组能够用来计算，否则出错  
④当输入数组的某个轴的长度为1时，沿着此轴运算时都用此轴上的第一组值

#### （2）ufunc的方法 ####

①reduce 方法和Python的reduce函数类似，它沿着axis轴对array进行操作，相当于将运算符插入到沿axis轴的所有子数组或者元素当中。  
②accumulate 方法和reduce方法类似，只是它返回的数组和输入的数组的shape相同，保存所有的中间计算结果。  
③reduceat 方法计算多组reduce的结果，通过indices参数指定一系列reduce的起始和终了位置。

## 3.矩阵运算 ##

（1）matrix对象  
numpy库提供了matrix类，使用matrix类创建的是矩阵对象，它们的加减乘除运算缺省采用矩阵方式计算，因此用法和matlab十分类似。但是由于NumPy中同时存在ndarray和matrix对象，因此用户很容易将两者弄混。这有违Python的“显式优于隐式”的原则，因此并不推荐在较复杂的程序中使用matrix。  
（2）计算乘积的函数  
①dot：对于两个一维的数组，计算的是这两个数组对应下标元素的乘积和(数学上称之为内积)；对于二维数组，计算的是两个数组的矩阵乘积；对于多维数组，它的通用计算公式如下，即结果数组中的每个元素都是：数组a的最后一维上的所有元素与数组b的倒数第二位上的所有元素的乘积和。

dot(a,b)[i,j,k,m]=sum(s[i,j,:]*b[k,:,m])

②inner：和dot乘积一样，对于两个一维数组，计算的是这两个数组对应下标元素的乘积和；对于多维数组，它计算的结果数组中的每个元素都是：数组a和b的最后一维的内积，因此数组a和b的最后一维的长度必须相同。

inner(a, b)[i,j,k,m] = sum(a[i,j,:]*b[k,m,:])

③outer：只按照一维数组进行计算，如果传入参数是多维数组，则先将此数组展平为一维数组之后再进行运算。outer乘积计算的列向量和行向量的矩阵乘积。  
④除此之外还有一些函数，如：inv函数计算逆矩阵，solve函数可以求解多元一次方程组。

## 4.文件存取 ##

NumPy提供了多种文件操作函数方便我们存取数组内容。文件存取的格式分为两类：二进制和文本。  
而二进制格式的文件又分为NumPy专用的格式化二进制类型和无格式类型。  
使用数组的方法函数tofile可以方便地将数组中数据以二进制的格式写进文件。tofile输出的数据没有格式，因此用numpy.fromfile读回来的时候需要自己格式化数据。

>>> a = np.arange(0,12)
    >>> a.shape = 3,4
    >>> a
    array([[ 0, 1, 2, 3],
    [ 4, 5, 6, 7],
    [ 8, 9, 10, 11]])
    >>> a.tofile("a.bin")
    >>> b = np.fromfile("a.bin", dtype=np.float) # 按照float类型读入数据
    >>> b # 读入的数据是错误的
    array([ 2.12199579e-314, 6.36598737e-314, 1.06099790e-313,
    1.48539705e-313, 1.90979621e-313, 2.33419537e-313])
    >>> a.dtype # 查看a的dtype
    dtype('int32')
    >>> b = np.fromfile("a.bin", dtype=np.int32) # 按照int32类型读入数据
    >>> b # 数据是一维的
    array([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11])
    >>> b.shape = 3, 4 # 按照a的shape修改b的shape
    >>> b # 这次终于正确了
    array([[ 0, 1, 2, 3],
    [ 4, 5, 6, 7],
    [ 8, 9, 10, 11]])

从上面的例子可以看出，需要在读入的时候设置正确的dtype和shape才能保证数据一致。并且tofile函数不管数组的排列顺序是C语言格式的还是Fortran语言格式的，统一使用C语言格式输出。

[70]: /images/20220519/0713abad2a8d43d882886d734f6d44ce.png
[70 1]: /images/20220519/c861af0a30f14668bf921e8e77528afe.png
[70 2]: /images/20220519/39f3fe6f22a14a8392b611eca4d489ec.png
[70 3]: /images/20220519/c2a849ae82584b7a9381a0f2fa48ca62.png