Numpy:数组合矢量计算

简介: Numpy, 数组和矢量计算包前几年前想学数据分析,于是就去学习Python的Numpy。然而看完《利用Python进行数据分析》后,也对它没有多大印象的。

Numpy, 数组和矢量计算包

前几年前想学数据分析,于是就去学习Python的Numpy。然而看完《利用Python进行数据分析》后,也对它没有多大印象的。但是学了一段时间R语言,并且将R语言和Python进行一些比较,再次学习Numpy就特别轻松了。

由于学过R语言,我可以简单认为Numpy提供的多维数据对象ndarray就是Python版本的R语言的vector, matrixarray。几乎没有特殊说明,这两者的任何操作都是一致的。

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吧。

目录
相关文章
|
3月前
|
存储 大数据 索引
【Python】NumPy数组和矢量计算
【1月更文挑战第26天】【Python】NumPy数组和矢量计算
|
Python
python计算的效率问题-pandas、numpy结合代替遍历pandas数据
python计算的效率问题-pandas、numpy结合代替遍历pandas数据
83 0
python计算的效率问题-pandas、numpy结合代替遍历pandas数据
python_numpy_计算对数收益率和还原问题
python_numpy_计算对数收益率和还原问题
102 0
|
Python
numpy向量计算
numpy向量计算
72 0
|
机器学习/深度学习 存储 Serverless
NumPy 与 Python 内置列表计算标准差的区别
NumPy,是 Numerical Python 的简称,用于高性能科学计算和数据分析的基础包,像数学科学工具(pandas)和框架(Scikit-learn)中都使用到了 NumPy 这个包。
|
数据挖掘 Python
Python | Numpy:详解计算矩阵的均值和标准差
对于 CRITIC 权重法而言,在标准差一定时,指标间冲突性越小,权重也越小;冲突性越大,权重也越大;另外,当两个指标间的正相关程度越大时,(相关系数越接近1),冲突性越小,这表明这两个指标在评价方案的优劣上反映的信息有较大的相似性。
323 0
Python | Numpy:详解计算矩阵的均值和标准差
|
算法 Python 计算机视觉
用numpy计算成交量加权平均价格(VWAP),并实现读写文件
VWAP(Volume-Weighted Average Price,成交量加权平均价格)是一个非常重要的经济学量,它代表着金融资产的“平均”价格。某个价格的成交量越高,该价格所占的权重就越大。VWAP就是以成交量为权重计算出来的加权平均值,常用于算法交易。
2576 0
|
机器学习/深度学习 Ubuntu Linux
谷歌开源计算框架JAX:比Numpy快30倍,还可在TPU上运行!
大家有了解过JAX吗?JAX是一种可在CPU、GPU和TPU上运行的“Numpy”,专门针对机器学习研究,并提供高性能自微分计算能力,速度要比纯用Numpy快几十倍!
404 0
谷歌开源计算框架JAX:比Numpy快30倍,还可在TPU上运行!
|
存储 数据挖掘 数据处理
《利用Python进行数据分析·第2版》第4章 NumPy基础:数组和矢量计算
第1章 准备工作第2章 Python语法基础,IPython和Jupyter第3章 Python的数据结构、函数和文件 第4章 NumPy基础:数组和矢量计算第5章 pandas入门第6章 数据加载、存储与文件格式第7章 数据清洗和准备第8章 数据规...
1236 0
|
1月前
|
机器学习/深度学习 存储 算法
Python中的NumPy库:数值计算与科学计算的基石
【2月更文挑战第29天】NumPy是Python科学计算的核心库,专注于高效处理大型多维数组和矩阵。其核心是ndarray对象,提供快速数组操作和数学运算,支持线性代数、随机数生成等功能。NumPy广泛应用于数据处理、科学计算和机器学习,简化了矩阵运算、统计分析和算法实现,是数据科学和AI领域的重要工具。