美文网首页我是程序员@IT·互联网互联网科技
NumPy快餐教程(1) - 如何生成多维数组

NumPy快餐教程(1) - 如何生成多维数组

作者: 阿里云云栖号 | 来源:发表于2018-05-14 16:45 被阅读213次

    摘要: 在算法中我们最经常用到的就是矩阵,我们就从矩阵开始说起吧。 NumPy中,使用二维的多维数组ndarray来存储矩阵。

    Python现在是最热门的人工智能语言,各种工具的支持如Google的Tensorflow,都是首选支持Python的。

    但是,与R语言不同,Python语言设计时,并没有考虑对于矩阵运算,统计计算等功能做专项支持。于是我们需要NumPy库来补足这一能力上的不足。

    NumPy是Python的著名扩展库,相当于Python中的MATLAB。

    如何生成多维数组

    初识ndarray多维数组

    在算法中我们最经常用到的就是矩阵,我们就从矩阵开始说起吧。

    NumPy中,使用二维的多维数组ndarray来存储矩阵。

    例:

    a3 = np.array([[1,0],[0,1]])

    会生成这样一个多维数组对象

    array([[1, 0],

    [0, 1]])

    生成数组序列

    通过开始值、结束值和步长值生成数组序列 - arange

    可以通过arange函数来生成指定开始值,结束值和步长值的一维数组。请注意,结束值并不包含在序列中,也就是说结束值是开区间。

    In [25]: a4 = np.arange(1,10,1)

    In [26]: a4

    Out[26]: array([1,2,3,4,5,6,7,8,9])

    线性序列 - linspace

    与arange类似,linspace通过给定初值、终值和元素个数来生成序列。是否包含终值可以通过endpoint属性来设置。

    例:

    In [37]: a8 = np.linspace(1,10,10,endpoint=True)

    In [38]: a8

    Out[38]: array([1.,2.,3.,4.,5.,6.,7.,8.,9.,10.])

    等比序列 - logspace

    除了线性的等差数列,我们也可以通过等比数列的方式来生成一维数组。

    默认是以10的n次方为参数,比如logspace(0,4,3)的意思是,初值为10的0次方,即1,终值是10的4次方,即100,一共生成3个值。

    例,生成[1,100,10000]

    In [47]: a9 = np.logspace(0,4,3)

    In [48]: a9

    Out[48]: array([1.00000000e+00,1.00000000e+02,1.00000000e+04])

    我们当然也可以修改基数,比如改成3:

    In [53]: a10 = np.logspace(1,5,3,base=3)

    In [54]: a10

    Out[54]: array([3.,27.,243.])

    改变多维数组的形状

    如果有一个一维数组要转为多维数组,可以通过修改shape属性来实现。

    我们可以先将数据存在一维数组中,可以用列表或者元组来生成一维数组,它们是等价的:

    例:

    In [2]: a1 = np.array([1,2,3,4])

    In [3]: a1

    Out[3]:array([1,2,3,4])

    In [4]: a2 = np.array((1,0,0,1))

    In [5]: a2

    Out[5]:array([1,0,0,1])

    我们通过shape属性来查看一个数组的形状:

    In [14]: a1.shape

    Out[14]: (4,)

    In [15]: a2.shape

    Out[15]: (4,)

    shape属性是可以直接修改的,比如我们想把上面的a1改成2 x 2的矩阵,就直接改shape值就是了:

    In [16]: a1.shape =2,2

    In [17]: a1

    Out[17]: array([[1,2],      [3,4]])

    如果能确定一个轴,另一个可以赋-1让系统自己去算。

    例:

    In [18]: a2.shape=2,-1

    In [19]: a2

    Out[19]: array([[1,0],      [0,1]])

    如果想保持这个数组不变,生成一个形状改变的新数组,可以调用reshape方法。

    例:我们将一个25个元素的数组生成一个5x5的新数组

    In [59]: a11 = np.linspace(1,100,25)

    In [60]: a11

    Out[60]: array([1.,5.125,9.25,13.375,17.5,21.625,25.75,29.875,34.,38.125,42.25,46.375,50.5,54.625,58.75,62.875,67.,71.125,75.25,79.375,83.5,87.625,91.75,95.875,100.])

    In [61]: a12 = a11.reshape(5,-1)

    In [62]: a12

    Out[62]:

    array([[1.,5.125,9.25,13.375,17.5],      [21.625,25.75,29.875,34.,38.125],      [42.25,46.375,50.5,54.625,58.75],      [62.875,67.,71.125,75.25,79.375],      [83.5,87.625,91.75,95.875,100.]])

    直接生成多维数组

    生成全0的数组

    zeros生成全是0的数组,第一个参数是shape

    例:

    In [65]: np.zeros((10,10))

    Out[65]:

    array([[0.,0.,0.,0.,0.,0.,0.,0.,0.,0.],      [0.,0.,0.,0.,0.,0.,0.,0.,0.,0.],      [0.,0.,0.,0.,0.,0.,0.,0.,0.,0.],      [0.,0.,0.,0.,0.,0.,0.,0.,0.,0.],      [0.,0.,0.,0.,0.,0.,0.,0.,0.,0.],      [0.,0.,0.,0.,0.,0.,0.,0.,0.,0.],      [0.,0.,0.,0.,0.,0.,0.,0.,0.,0.],      [0.,0.,0.,0.,0.,0.,0.,0.,0.,0.],      [0.,0.,0.,0.,0.,0.,0.,0.,0.,0.],      [0.,0.,0.,0.,0.,0.,0.,0.,0.,0.]])

    生成全是1的数组

    例:

    In [66]: np.ones((5,5))

    Out[66]:

    array([[1.,1.,1.,1.,1.],      [1.,1.,1.,1.,1.],      [1.,1.,1.,1.,1.],      [1.,1.,1.,1.,1.],      [1.,1.,1.,1.,1.]])

    只生成空数组

    empty不赋初值,是最快速的方法

    例:

    In [67]: np.empty((3,3))

    Out[67]: array([[1.,2.125,3.25],      [4.375,5.5,6.625],      [7.75,8.875,10.]])

    通过函数来生成数组

    通过fromfunction函数可以通过一个函数来生成想要的数组。

    例,生成九九乘法表:

    In [125]:defmul2(x,y):...:return(x+1)*(y+1)    ...:

    In [126]: np.fromfunction(mul2,(9,9))

    Out[126]:

    array([[1.,2.,3.,4.,5.,6.,7.,8.,9.],      [2.,4.,6.,8.,10.,12.,14.,16.,18.],      [3.,6.,9.,12.,15.,18.,21.,24.,27.],      [4.,8.,12.,16.,20.,24.,28.,32.,36.],      [5.,10.,15.,20.,25.,30.,35.,40.,45.],      [6.,12.,18.,24.,30.,36.,42.,48.,54.],      [7.,14.,21.,28.,35.,42.,49.,56.,63.],      [8.,16.,24.,32.,40.,48.,56.,64.,72.],      [9.,18.,27.,36.,45.,54.,63.,72.,81.]])

    本文作者:lusing

    阅读原文

    本文为云栖社区原创内容,未经允许不得转载。

    相关文章

      网友评论

        本文标题:NumPy快餐教程(1) - 如何生成多维数组

        本文链接:https://www.haomeiwen.com/subject/hkyddftx.html