首页 > 技术文章 > Py中的多维数组ndarray学习【转载】

BlueBlueSea 2019-01-10 21:42 原文

转自:http://blog.sciencenet.cn/home.php?mod=space&uid=3031432&do=blog&id=1064033

1. NumPy中的N维数组ndarray基本介绍

- NumPy中基本的数据结构

- 所有元素是同一种类型

- 别名array(数组)

- 节省内存,提高CPU计算时间

- 有丰富的函数

注:NumPy的思维模式是面向数组

2.ndarray数组属性

- 下标从0开始。

- 一个ndarray数组中的所有元素的类型必须相同。

- 轴(axis):每一个线性的数组称为是一个轴,也就是维度(dimensions)。比如,二维数组相当于是两个一维数组,其中第一个一维数组中每个元素又是一个一维数组,所以一维数组就是ndarray中的轴,第一个轴(也就是第0轴)相当于是底层数组,第二个轴(也就是第1轴)是底层数组里的数组

很多时候可以声明axis。axis=0,表示沿着第0轴进行操作,即对每一列进行操作;axis=1,表示沿着第1轴进行操作,即对每一行进行操作。

- 秩(rank):维数,一维数组的秩为1,二维数组的秩为2,以此类推。即轴的个数。

- 基本属性

ndarray1.ndim - 秩

ndarray1.shape - 维度 # 是一个元组,表示数组在每个维度上的大小。比如,一个二维数组,其维度表示“行数”和“列数”。该元组的长度即为秩

ndarray1.size - 元素总个数 # 等于shape属性中元组元素的乘积。

ndarray1.dtype - 元素类型

ndarray1.itemsize - 元素字节大小 # 即元素所占内存空间大小,例如,元素类型为float64的数组的itemsiz属性值为8(=64/8),元素类型为complex32的数组的itemsize属性为4(=32/8)。

ndarray1.data - 包含实际数组元素的缓冲区,由于一般通过数组的索引获取元素,所以通常不需要使用这个属性。

例如:以下5x5的数组,可以看成由5个一维数组构成,每个一维数组包含5个元素;第一维被称为第0轴(列),第二维被称为第1轴(行);秩为2,维度为(5,5),元素总个数为25。

3.如何创建ndarray

3.1从已有数据-——从list、tuple

array() - 创建多维数组。

np.array(object, dtype=None, copy=True, order='K', subok=False, ndmin=0)

object — list或tuple对象。强制参数。

dtype — 数据类型。可选参数。

copy — 默认为True,对象被复制。可选参数。

order — 数组按一定的顺序排列。C - 按行;F - 按列;A - 如果输入为F则按列排列,否则按行排列;K - 保留按行和列排列。默认值为K。可选参数。

subok — 默认为False,返回的数组被强制为基类数组。如果为True,则返回子类。可选参数。

ndmin — 最小维数。可选参数。

注:array函数的参数必须是由方括号括起来的列表,而不能使用多个数值作为参数调用array。

 

>>> import numpy as np
>>> arr1=np.array([[1,2,3],[4.1,5.1,6.1]])
>>> arr1
array([[1. , 2. , 3. ],
       [4.1, 5.1, 6.1]])
>>> arr2=np.array([(1,2,3),(4.1,5.1,6.1)])
>>> arr2
array([[1. , 2. , 3. ],
       [4.1, 5.1, 6.1]])

 

//上述代码中1是由元祖创建,2是由列表创建。

3.2从字符串中读取 - fromstring()

fromstring() - 从字符串中读取数据,并将其转换为一维数组。

np.fromstring(string, dtype=float, count=-1, sep='')

string — 包含数据的字符串。强制参数。

dtype — 数据类型,默认为浮点型。可选参数。

count — 从左到右读取数据的个数。默认为-1,表示读取所有数据。可选参数。

sep — 分隔符。若不指定分隔符,或指定为空,则字符串包含的数据被解译为二进制数据,否则为带有小数的ASCII文本。可选参数。

>>> arr1=np.fromstring('1,2,3,4',dtype=int,sep=',')
>>> arr1
array([1, 2, 3, 4])
>>> arr2=np.fromstring('1 2 3 4',dtype=int,sep=' ')
>>> arr2
array([1, 2, 3, 4])

 

 3.3从可迭代对象中读取 - fromiter()

fromiter() - 从可迭代对象中读取数据,并将其转换为一维数组。

np.fromiter(iterable, dtype, count=-1)

iterable — 可迭代对象。不能有嵌套。强制参数。

dtype — 数据类型。强制参数。

count — 表示从可迭代对象中读取的项目数。默认为-1,表示读取所有数据。可选参数。

>>> it=(x*x for x in range(5))
>>> it
<generator object <genexpr> at 0x000001B85AB17BA0>

>>> x*x for x in range(5)
SyntaxError: invalid syntax
>>> np.fromiter(it,float)
array([ 0.,  1.,  4.,  9., 16.])
>>> np.fromiter('1234',int) array([1, 2, 3, 4])
>>> np.fromiter('1,2,3,4',int)#这样对于int是不可迭代的 Traceback (most recent call last): File "<pyshell#14>", line 1, in <module> np.fromiter('1,2,3,4',int) ValueError: invalid literal for int() with base 10: ','

>>> np.fromiter((1,2,3,4),float) array([1., 2., 3., 4.])

 

3.4以函数式创建 - fromfunction()

fromfunction() - 在每个坐标轴上执行函数表达式,用得到的数据创建数组。

np.fromfunction(function, shape, dtype)

function — 可调用的函数。必选参数。

shape — 要创建的数组的维度。其长度与函数参数的个数一致。必选参数。

dtype — 数据类型,默认为浮点型。可选参数

//这里的x y,我是真的不太理解。为什么就得到这个了?

 

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

3.5创建全1数组—— ones(), ones_like() 

ones() - 创建给定形状的ndarray,并将数组中所有元素填充为1。

np.ones(shape, dtype=None, order='C')

shape — 维度。强制参数。

dtype — 数据类型。默认为np.float64。可选参数。

order — 数组按一定的顺序排列。C - 按行;F - 按列。默认值为C。可选参数。

ones_like() -  创建与参数数组形状和类型相同的ndarray,并将数组中所有元素填充为1。

np.ones_like(a, dtype=None, order='K', subok=True)

a — 给定的数组。其维度和数据类型决定着要创建的数组的属性。强制参数。

dtype — 数据类型。默认的“None”意为“a.dtype”。可选参数。

order — 数组按一定的顺序排列。C - 按行;F - 按列;A - 如果输入为F则按列排列,否则按行排列;K - 保留按行和列排列。默认值为K。可选参数。

subok — 默认为True,返回的数组被强制为子类数组。如果为False,则返回基类数组。可选参数。

>>> x=np.ones((5))
>>> x.ndim
1
>>> x.shape
(5,)
>>> np.ones((5,2),int)#参数shape
array([[1, 1],
       [1, 1],
       [1, 1],
       [1, 1],
       [1, 1]])
>>> a=np.array([[1,2,3],[4,5,6]])
>>> np.ones_like(a)
array([[1, 1, 1],
       [1, 1, 1]])
>>> np.ones((5,))#这两个都是创建一维的数组
array([1., 1., 1., 1., 1.])
>>> np.ones((5))#这两个都是创建一维的数组
array([1., 1., 1., 1., 1.])

 

3.6创建‘全0’数组 - zeros(), zeros_like()

zeros() - 创建给定形状的ndarray,并将数组中所有元素填充为0。

np.zeros(shape, dtype=float, order='C')

shape — 维度。强制参数。

dtype — 数据类型。默认为np.float64。可选参数。

order — 数组按一定的顺序排列。C - 按行;F - 按列。默认值为C。可选参数。

zeros_like() -  创建与参数数组形状和类型相同的ndarray,并将数组中所有元素填充为0。

np.zeros_like(a, dtype=None, order='K', subok=True)

a — 给定的数组。其维度和数据类型决定着要创建的数组的属性。强制参数。

dtype — 数据类型。默认的“None”意为“a.dtype”。可选参数。

order — 数组按一定的顺序排列。C - 按行;F - 按列;A - 如果输入为F则按列排列,否则按行排列;K - 保留按行和列排列。默认值为K。可选参数。

subok — 默认为True,返回的数组被强制为子类数组。如果为False,则返回基类数组。可选参数。

>>> np.zeros((5,2))
array([[0., 0.],
       [0., 0.],
       [0., 0.],
       [0., 0.],
       [0., 0.]])
>>> np.zeros_like(a)
array([[0, 0, 0],
       [0, 0, 0]])
#与创建全1数组相似

 

3.7创建空数组 - empty(), empty_like()

empty() - 返回一个没有初始化内存的数组,该数组是空的。

np.empty(shape, dtype=float, order='C')

shape — 维度。强制参数。

dtype — 数据类型。默认为np.float64。可选参数。

order — 数组按一定的顺序排列。C - 按行;F - 按列。默认值为C。可选参数。

empty_like() - 返回一个与给定数组的维度和数据类型相同的新数组。

np.empty_like(a, dtype=None, order='K', subok=True)

a — 给定的数组。其维度和数据类型决定着要创建的数组的属性。强制参数。

dtype — 数据类型。默认的“None”意为“a.dtype”。可选参数。

order — 数组按一定的顺序排列。C - 按行;F - 按列;A - 如果输入为F则按列排列,否则按行排列;K - 保留按行和列排列。默认值为K。可选参数。

subok — 默认为True,返回的数组被强制为子类数组。如果为False,则返回基类数组。可选参数。

>>> np.zeros((5,2))#创建0数组
array([[0., 0.],
       [0., 0.],
       [0., 0.],
       [0., 0.],
       [0., 0.]])
>>> np.zeros_like(a)
array([[0, 0, 0],
       [0, 0, 0]])
>>> np.full((2,3),4)#用数据填充
array([[4, 4, 4],
       [4, 4, 4]])
>>> np.full_like(a,2)
array([[2, 2, 2],
       [2, 2, 2]])
>>> np.empty((2,2))
array([[6.01347002e-154, 4.02456140e-066],
       [5.56435215e+141, 4.48505083e-308]])
>>> np.empty((2,2),int)#创建空数组,不会申请空间
array([[0, 0],
       [0, 0]])
>>> np.eye((2,3))#eye创建对角矩阵,它与其他的参数是不一样的,参数分别是 行,列。
Traceback (most recent call last):
  File "<pyshell#32>", line 1, in <module>
    np.eye((2,3))
  File "C:\Users\85937\AppData\Local\Programs\Python\Python36\lib\site-packages\numpy\lib\twodim_base.py", line 186, in eye
    m = zeros((N, M), dtype=dtype, order=order)
TypeError: 'tuple' object cannot be interpreted as an integer
>>> np.eye((2,3)
       
SyntaxError: invalid character in identifier
>>> np.eye(2,3)
       
array([[1., 0., 0.],
       [0., 1., 0.]])
>>> np.identity(3)#创建单位矩阵,只有一个整型,创建的是方阵。
       
array([[1., 0., 0.],
       [0., 1., 0.],
       [0., 0., 1.]])
>>> 

 

4.改变数组形状

reshape() - 改变数组的形状。通过reshape生成的新数组和原始数组共用一个内存,也就是说,若更改其中数组的元素,另一个数组也将发生改变。

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

ndarray1.reshape(newshape)

>>> a.reshape(3,2)#即当前数组的reshape函数
       
array([[1, 2],
       [3, 4],
       [5, 6]])
>>> a.reshape(3,2,order='F')#参数改为F,按列。不太明白这个是如何搞的
       
array([[1, 5],
       [4, 3],
       [2, 6]])
>>> a
       
array([[1, 2, 3],
       [4, 5, 6]])
>>> b=a.reshape(3,2)
       
>>> b
       
array([[1, 2],
       [3, 4],
       [5, 6]])
>>> b[2,1]=0#修改变形之后的b中的元素值,同样会改变a,两者共用的是一个存储空间
       
>>> a
       
array([[1, 2, 3],
       [4, 5, 0]])
>>> b
       
array([[1, 2],
       [3, 4],
       [5, 0]])
>>> np.reshape(a,2,3)
       
array([[1, 2, 3],
       [4, 5, 0]])
>>> b=np.reshape(a,(2,3))
       
>>> b
       
array([[1, 2, 3],
       [4, 5, 0]])
>>> b=np.reshape(a,(3,2))
       
>>> b
       
array([[1, 2],
       [3, 4],
       [5, 0]])
>>> b[2,1]=6#np.reshape效果也是一样的。
       
>>> b
       
array([[1, 2],
       [3, 4],
       [5, 6]])
>>> a
       
array([[1, 2, 3],
       [4, 5, 6]])

 

5创建等差数列数组 

arange() - 创建一个一维的等差数列数组,与Python中的range()函数类似。区别在于,np.arange()返回的是一个numpy数组,而Python中的range()函数返回的是一个列表。

np.arange(start, stop, step步长, dtype=None)//这里的stop是取不到的

 

linspace() - 创建一个一维数组,在给定的区间上num等分

np.linspace(start, stop, num=50, endpoint=True, retstep=False, dtype=None)

stop — 区间终止值(是否取得到,需要设定参数endpoint)。强制参数。

>>> np.arange(3)
       
array([0, 1, 2])#取不到3
>>> range(3)
       
range(0, 3)#是一个list
>>> np.arange(1,5,2)
       
>>> np.arange(1,3.1,2)
       
array([1., 3.])
>>> np.linspace(1,5,2)
       
array([1., 5.])
>>> np.linspace(1,5,5)
       
array([1., 2., 3., 4., 5.])

 

6.数据切片与索引

一维数组

二维数组

//以切片方式索引,与 索引。

以数组索引:

第6行中的意思是A的[0,1]和[2,3]元素是什么?

第8行的意思是B中的[0,0]和[1,1]元素是什么?

也就是说前后两个数组长度一样,是两两对应的。

布尔型索引

神奇的布尔索引。

 

>>> B=np.arange(10).reshape(2,5)
       
>>> B
       
array([[0, 1, 2, 3, 4],
       [5, 6, 7, 8, 9]])
>>> for row in B.T:
       print(row)

       
[0 5]
[1 6]
[2 7]
[3 8]
[4 9]
>>> for col in B.T:
       print(col)

       
[0 5]
[1 6]
[2 7]
[3 8]
[4 9]
>>> B.T
       
array([[0, 5],
       [1, 6],
       [2, 7],
       [3, 8],
       [4, 9]])

 

//这里对B的转置中,遍历行和遍历列都是一样的结果。 

 列出B中的元素:

转化为一维数组之后打印。B并未改变

最值索引

in5行中,参数1表示按行,in7中有0,表示按列索引。

注意返回的不是最值,而是最值的索引即下标。 

 7.ndarray的运算

加减乘除、点乘、乘方、整余数

以上的运算,在np中有相应的函数。

点乘就是矩阵乘法。

推荐阅读