Python数据分析 | Numpy与1维数组操作

虚幻大学 xuhss 467℃ 0评论

Python微信订餐小程序课程视频

https://edu.csdn.net/course/detail/36074

Python实战量化交易理财系统

https://edu.csdn.net/course/detail/35475
watermark,type d3F5LXplbmhlaQ,shadow 50,text Q1NETiBAU2hvd01lQUk=,size 20,color FFFFFF,t 70,g se,x 16#pic center - Python数据分析 | Numpy与1维数组操作

作者:韩信子@ShowMeAI
教程地址http://www.showmeai.tech/tutorials/33
本文地址http://www.showmeai.tech/article-detail/142
声明:版权所有,转载请联系平台与作者并注明出处


watermark,type d3F5LXplbmhlaQ,shadow 50,text Q1NETiBAU2hvd01lQUk=,size 20,color FFFFFF,t 70,g se,x 16 - Python数据分析 | Numpy与1维数组操作

n维数组是NumPy的核心概念,大部分数据的操作都是基于n维数组完成的。本系列内容覆盖到1维数组操作2维数组操作3维数组操作方法,本篇讲解Numpy与1维数组操作。

一、向量初始化

可以通过Python列表创建NumPy数组。

135d35ff9022b7c43d99f38a5a54f157 - Python数据分析 | Numpy与1维数组操作

如图中(a),将列表元素转化为一维数组。注意,这里一般会确保列表元素类型相同,否则默认dtype=’object',会影响后续运算,甚至产生语法错误。

由于在数组末尾没有预留空间以快速添加新元素,NumPy数组无法像Python列表那样增长。因此,通常的处理方式包括:

  • 在变长Python列表中准备好数据,然后将其转换为NumPy数组
  • 使用 np.zerosnp.empty 预先分配必要的空间(图中b)

通过图中(c)方法,可以创建一个与某一变量形状一致的空数组。

不止是空数组,通过上述方法还可以将数组填充为特定值:

460548c79ec1dec2f5d279ac0da368c4 - Python数据分析 | Numpy与1维数组操作

在NumPy中,还可以通过单调序列初始化数组:

9e66958ddf9aedbc9970ba91bf4f19f9 - Python数据分析 | Numpy与1维数组操作

如果我们需要浮点数组,可以使用 arange(3).astype(float) 这样的操作更改arange输出的类型,也可以在参数端使用浮点数,比如 arange(4.) 来生产浮点数Numpy数组。
以下是arange浮点类型数据可能出现的一些问题及解决方案:

465f18ed8144572556cdedf6a50b1aab - Python数据分析 | Numpy与1维数组操作

图中,0.1对我们来说是一个有限的十进制数,但对计算机而言,它是一个二进制无穷小数,必须四舍五入为一个近似值。因此,将小数作为arange的步长可能导致一些错误。可以通过以下两种方式避免如上错误:

  • 使间隔末尾落入非整数步数,但这会降低可读性和可维护性;
  • 使用linspace,这样可以避免四舍五入的错误影响,并始终生成要求数量的元素。
    • 使用linspace时尤其需要注意最后一个的数量参数设置,由于它计算点数量,而不是间隔数量,因此上图中数量参数是11,而不是10。

随机数组的生成方法如下:

f3e297b7c000b944cafda3adbea32ff9 - Python数据分析 | Numpy与1维数组操作

二、向量索引

NumPy可以使用非常直接的方式对数组数据进行访问:

4673ca066107ab41f3dc3ca2bdf476ee - Python数据分析 | Numpy与1维数组操作

图中,除“fancy indexing”外,其他所有索引方法本质上都是views:它们并不存储数据,如果原数组在被索引后发生更改,则会反映出原始数组中的更改。

上述所有这些方法都可以改变原始数组,即允许通过分配新值改变原数组的内容。这导致无法通过切片来复制数组。如下是python列表和NumPy数组的对比:

67935bd86f8c8f90454d11e735e27e63 - Python数据分析 | Numpy与1维数组操作

NumPy数组支持通过布尔索引获取数据,结合各种逻辑运算符可以有很高级的数据选择方式,这在Python列表中是不具备的:

cf537335685ed728fad4f65275ed79cd - Python数据分析 | Numpy与1维数组操作

注意,不可以使用3 <= a <= 5这样的Python“三元”比较。

如上所述,布尔索引是可写的。如下图 np.wherenp.clip 两个专有函数。

df268e8a10d15effd1969cf21bb16a49 - Python数据分析 | Numpy与1维数组操作

三、向量操作

NumPy支持快速计算,向量运算操作接近C++速度级别,并不受Python循环本身计算慢的限制。NumPy允许像普通数字一样操作整个数组:

cbc7f0a8b8f2455bcc96a2420de89ac2 - Python数据分析 | Numpy与1维数组操作

在python中,a//b表示a div b(除法的商),x**n表示 xⁿ

浮点数的计算也是如此,NumPy能够将标量广播到数组:

3da2c98797a17c2fad7940234326959a - Python数据分析 | Numpy与1维数组操作

Numpy提供了许多数学函数来处理矢量:

ee82eaf883bcdd7fe7c9cd62e5eda4e0 - Python数据分析 | Numpy与1维数组操作

向量点乘(内积)和叉乘(外积、向量积)如下:

66cd74acc149ec244c69ea7cd0241102 - Python数据分析 | Numpy与1维数组操作

NumPy也提供了如下三角函数运算:

ceb4252242272e245f4497110de906f4 - Python数据分析 | Numpy与1维数组操作

数组整体进行四舍五入:

5118c91f8fe9ae24d829a25a22fc4c3d - Python数据分析 | Numpy与1维数组操作

floor向上取整,ceil向下取整,round四舍五入

np.aroundnp.round 是等效的,这样做只是为了避免 from numpy import * 时与Python around的冲突(但一般的使用方式是import numpy as np)。当然,你也可以使用a.round()。

NumPy还可以实现以下功能:

bea57a93e647d0c09a8164fd126fe91c - Python数据分析 | Numpy与1维数组操作

以上功能都存在相应的nan-resistant变体:例如nansumnanmax

在NumPy中,排序函数功能有所阉割:

c9e8709b42dbd571dc817634ce079584 - Python数据分析 | Numpy与1维数组操作

对于一维数组,可以通过反转结果来解决reversed函数缺失的不足,但在2维数组中该问题变得棘手。

四、查找向量中的元素

NumPy数组并没有Python列表中的索引方法,索引数据的对比如下:

1000f4644dcfd88382087f97b6425923 - Python数据分析 | Numpy与1维数组操作

index()中的方括号表示 j 或 i&j 可以省略

  • 可以通过 np.where(a==x)[0] [0]查找元素,但这种方法很不pythonic,哪怕需要查找的项在数组开头,该方法也需要遍历整个数组。
  • 使用Numba实现加速查找,next((i[0] for i, v in np.ndenumerate(a) if v==x), -1),在最坏的情况下,它的速度要比where慢。
  • 如果数组是排好序的,使用v = np.searchsorted(a, x); return v if a[v]==x else -1时间复杂度为O(log N),但在这之前,排序的时间复杂度为O(N log N)。

实际上,通过C实现加速搜索并不是困难,问题是浮点数据比较。

五、浮点数比较

np.allclose(a, b)用于容忍误差之内的浮点数比较。

6ee315057b5710e6672ef36362b4c97f - Python数据分析 | Numpy与1维数组操作

  • np.allclose假定所有比较数字的尺度为1。如果在纳秒级别上,则需要将默认atol参数除以1e9:np.allclose(1e-9,2e-9, atol=1e-17)==False
  • math.isclose不对要比较的数字做任何假设,而是需要用户提供一个合理的abs_tol值(np.allclose默认的atol值1e-8足以满足小数位数为1的浮点数比较,即math.isclose(0.1+0.2–0.3, abs_tol=1e-8)==True

此外,对于绝队偏差和相对偏差,np.allclose依然存在一些问题。例如,对于某些值a、b, allclose(a,b)!=allclose(b,a),而在math.isclose中则不存在这些问题。查看GitHub上的浮点数据指南和相应的NumPy问题了解更多信息。

资料与代码下载

本教程系列的代码可以在ShowMeAI对应的github中下载,可本地python环境运行,能访问Google的宝宝也可以直接借助google colab一键运行与交互操作学习哦!

本系列教程涉及的速查表可以在以下地址下载获取:

拓展参考资料

ShowMeAI相关文章推荐

ShowMeAI系列教程推荐

e23a24cd1ca88522b1c8395a65cfde52 - Python数据分析 | Numpy与1维数组操作

转载请注明:xuhss » Python数据分析 | Numpy与1维数组操作

喜欢 (0)

您必须 登录 才能发表评论!