NumPy 学习 第一篇:创建数组和数组变形

NumPy是Python中用于科学计算的基础软件包,专门用于处理矩阵,数据类型是数值型的,用于对数值数据进行快速的计算。

NumPy提供了多维数据组对象,包中最核心的类型是ndarray,封装了python原生的 n 维数组,数组元素的类型是相同的,能够用于执行矢量算术运算,无需编写循环结构,就能对整个数组进行批量运算。在通常情况下,需要导入NumPy,并设置别名为np。

import numpy as np 

Numpy的array class 叫做ndarray,别名是array,ndarray对象有如下特征:

  • ndarray对象在创建时有固定的大小,更改ndarray的大小将创建一个新数组并删除原始数组。
  • ndarray对象中的元素类型是相同的,在内存中的大小相同。
  • ndarry有助于对大量数据进行高级数学运算和其他类型的运算。

一,创建ndarray

ndarray是N-Dimension-Array的简称,该对象是一个快速而灵活的大数据集容器,该容器中存储的元素的数据类型都是相同的。

创建数组的方法通常有:

  • 从其他Python结构(例如,列表,元组)转换
  • 通过增长序列来创建数组
  • 通过sharp或value(例如,ones、zeros等)创建

1,从已有的array-like对象(list,tuple)转换为Numpy数组

在Python中排列成array-like结构的数值数据可以通过使用array()函数转换为数组,例如列表和元组。

a1 = np.array([1,2,3,4,5])  #1row * 5col
a2 = np.array([[1,2,3,4,5]
         ,[6,7,8,9,10]])# 2row * 5col 

通过array()函数,使用列表创建的是一维数组,使用嵌套的列表创建的是多维数组。

numpy.array()函数的定义:

numpy.array(object, dtype=None)

参数注释:

object:array-like的数据对象,例如list,tuple

dtype:数组元组的类型,如果没有指定,那么是哦那个最小size的类型来存储数据。

2,通过增长序列来创建数组

通过arange()函数创建一维数组,数组的元素是一个序列,默认值start=0,不包括stop,step=1。

numpy.arange([start, ]stop, [step, ]dtype=None)

numpy.arange()的用法示例:

a = np.arange(10) #default start=0, end=10(exclude 10),step=1
# [0 1 2 3 4 5 6 7 8 9]
a1 = np.arange(5,10) # start=5, end=10(exclude=10),step=1
# [5 6 7 8 9]
a2 = np.arange(5,20,2) # start=5, end=20(exclude 20),step=2
#[ 5  7  9 11 13 15 17 19]

可以使用reshape()函数重塑ndarray数组的shape,把12个元素的一维数组转换为3行4列的二维数组:

>>> np.arange(12).reshape(3,4)
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])

3,通过内置的函数来创建数组

np.eye(n,m):用于创建n row * m col的数组,下标相同的元素值是1,下标不同的元素值是0;如果省略m,那么默认值创建的是n * n的方阵。

>>> np.eye(3)
array([[1., 0., 0.],
       [0., 1., 0.],
       [0., 0., 1.]])

np.identity(n):创建单位阵,维数由n来决定,相当于np.eye(n)。

np.diag(v):用于创建对角阵,对角阵是一个方阵,对角线上的元素值由参数v来指定:

>>> np.diag([1, 2, 3], 1)
array([[0, 1, 0, 0],
       [0, 0, 2, 0],
       [0, 0, 0, 3],
       [0, 0, 0, 0]])

np.zeros(shape):用于创建0阵,shape参数是int或tuple,用于指定矩阵的shape,

np.zeros((2,3))
array([[ 0.,  0.],
       [ 0.,  0.]])

np.ones(shape):用于创建元素值都为1的矩阵,shape参数是int或tuple,用于指定矩阵的shape,

np.ones((2,2))
array([[1.,  1.],
       [1.,  1.]])

二,ndarray的属性

数组对象的属性主要有:

  • dtype:描述数组元素的类型
  • shape:以tuple表示的数组形状
  • ndim:数组的维度
  • size:数组中元素的个数

1,dtype对象

dtype是一个特殊的对象,它表示数组元素的类型。下面罗列出NumPy内置的跟平台无关的常用的数据类型:

  • 整数有符号系列:np.int8,np.int16,np.int32,np.int64
  • 整数无符号系列:np.uint8、np.uint16、np.uint32、np.uint64
  • 浮点数:np.float32、np.float64
  • 布尔值:np.bool

2,shape对象

以元组形式表示一个数组的维度,前三个维度有名称:第一个维度是行,第二个维度是列,第三个维度是高

3,ndim对象

数组各个维度的元素数量

三,NumPy的数据类型

dtype(数据类型)是一个特殊的对象,具有name等属性,每一个ndarray对象都有一个dtype对象,可以通过astype()函数强制转换数据元素的类型:

>>> arr=np.array([1,2,3])
>>> arr.dtype
dtype('int32')
>>> arr.astype(np.uint8)
array([1, 2, 3], dtype=uint8)

Numpy支持的标量数据类型非常繁杂,我比较喜欢固定size的数据类型,因为固定size的数据类型跟平台无关,且命名格式比较规范,格式是:[u]type[size],u代表无符号,size代表占用的内存空间。

1,布尔类型

  • numpy.bool8

2,有符号的整数类型:

  • numpy.int8
  • numpy.int16
  • numpy.int32
  • numpy.int64

3,无符号的整数类型:

  • numpy.uint8
  • numpy.uint16
  • numpy.uint32
  • numpy.uint64

4,浮点数类型:

  • numpy.float16
  • numpy.float32
  • numpy.float64
  • numpy.float96
  • numpy.float128

5,复数类型:

  • numpy.complex64
  • numpy.complex128
  • numpy.complex192
  • numpy.complex256

6,日期时间类型

NumPy支持的日期时间类型是:np.datetime64,日期单位是年('Y'),月('M'),周('W')和天('D'), 而时间单位是小时('h'),分钟('m') ),秒('s'), 毫秒('ms')和一些额外的SI前缀基于秒的单位。 对于“非时间”值,datetime64数据类型还接受字符串“NAT”(不是时间), 以小写/大写字母的任意组合。

np.datetime64('2005-02-25')

从字符串创建日期时间数组时,仍然可以通过使用具有通用单位的日期时间类型从输入中自动选择单位。

>>> np.array(['2007-07-13', '2006-01-13', '2010-08-13'], dtype='datetime64')
array(['2007-07-13', '2006-01-13', '2010-08-13'], dtype='datetime64[D]')

np.arange()函数可以根据时间单位生成时间范围:

>>> np.arange('2005-02', '2005-03', dtype='datetime64[D]')

NumPy允许两个Datetime值相减,这个操作产生一个带有时间单位的数字。timedelta64的参数是一个数字(用于表示单位数),以及日期/时间单位,如 (D)ay, (M)onth, (Y)ear, (h)ours, (m)inutes, 或者 (s)econds。timedelta64数据类型也接受字符串“NAT”代替“非时间”值的数字。

>>> numpy.timedelta64(1, 'D')

Datetimes 和 Timedeltas 一起工作,为简单的日期时间计算提供方法。

>>> np.datetime64('2009-01-01') - np.datetime64('2008-01-01')
numpy.timedelta64(366,'D')
>>> np.datetime64('2009') + np.timedelta64(20, 'D')
numpy.datetime64('2009-01-21')

四,数组变形

一个数组的形状是由每个轴的元素数量决定的,可以通过reshape()和resize()函数来操纵数组的形状。transpose()函数用于数组的转置,ravel()函数用于把数组展开为一维数组,而flatten()用于获得数组展开之后的一个副本;swapaxes()函数用于交换数组的两个轴,squeeze()函数用于删除数组中长度为1的轴。

数组的转置可以通过数组的属性T来获得,数组的展开可以通过数组的属性flat来获得,但是flat是一个迭代器。

1,reshape函数

reshape()函数返回一个给定shape的数组的副本,不会修改原始数组,并且新的数组的size必须跟原始的数组size相兼容:

numpy.reshape(a, newshape, order='C')

参数 newshape 表示数组的形状,对于二维数组,newshape是 (row, col) ,第一个维度是行数,第二个维度是列数。

例如,下面的代码把一个一维数组转换为4行2列的二位数组:

a=np.arange(8)
a.reshape((4,2))

2,resize函数

resize()函数会修改原始数组,不会返回任何数据,直接对原始数组进行修改:

ndarray.resize(new_shape, refcheck=True)

参数new_shape是元组或n个int数字,表示数组的形状,即各个维度的元素数量,如果参数new_shape指定的size大于现有数组的size,那么新数组会增加新的元素,新增加的元素值都是0

>>> a.resize((2,6))
>>> a
array([[ 2.,  8.,  0.,  6.,  4.,  5.],
       [ 1.,  1.,  8.,  9.,  3.,  6.]])

3,展开数组

ravel()用于返回数组的展开形式,在展开成一维数组时,最右边的索引“变化最快”。

ndarray.ravel()
ndarray.flatten()

举个例子,从行维度和列维度的最小值开始,依次增加,获取的元素依次排列,构成一维数组:

>>> x = np.array([[1, 2, 3], [4, 5, 6]])
>>> np.ravel(x)
array([1, 2, 3, 4, 5, 6])

5,转置

x.transpose()
x.T

6,交换轴

把数组的两个轴相互交换

numpy.swapaxes(a, axis1, axis2)

7,移除长度为1的轴

>>> x = np.array([[[0], [1], [2]]])
>>> x.shape
(1, 3, 1)
>>> np.squeeze(x).shape
(3,)
>>> np.squeeze(x, axis=0).shape
(3, 1)

五,数组的转换

1,转换为list

ndarray对象的tolist()函数可以把一个数组对象转化为list列表:

>>> np.arange(12).reshape(3,4).tolist()
[[0, 1, 2, 3], [4, 5, 6, 7], [8, 9, 10, 11]]

2,修改元素的类型

调用astype()函数会创建一个新的数组,是原始数组的一个深拷贝。

arr.astype(np.int8)

3,创建数组的复制

数组的复制主要存在以下三种情况:

  • 直接赋值:无拷贝,相当于是引用
  • view():建立视图,浅拷贝,形状可以不一致,但数据相同,对于赋值和view操作,两个数组的数据是共享的,即引用同一个对象。
  • copy():深拷贝,完全独立的对象

数组的视图意味着数据不会被复制,视图上的任何修改都会直接反应到原始数组上。

deep_copy_arr = arr.copy()
view_arr = arr.view()

六,NumPy的常量

  • np.Inf 表示正无穷大
  • np.NINF 标识负无穷大
  • np.nan 表示非数字
  • np.pi 
  • np.e

 

参考文档:

Numpy Quickstart tutorial

作者悦光阴
本文版权归作者和博客园所有,欢迎转载,但未经作者同意,必须保留此段声明,且在文章页面醒目位置显示原文连接,否则保留追究法律责任的权利。
posted @ 2019-04-03 22:49  悦光阴  阅读(3182)  评论(0编辑  收藏  举报