numpy 多维数组的存取

Posted 曹孟德

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了numpy 多维数组的存取相关的知识,希望对你有一定的参考价值。

  多维数组的存取和一维数组类似,由于多维数组有多个轴,所以他的下标需要多个值来表示。这里讨论的主要是二维数组。二维数组0轴以行为单位,1轴以列为单位,存取数组使用元组作为下标,需要注意的是,python中的元组通常用圆括号括起来,但是其实元组的语法只需要用逗号隔开就可以。因此a[1,2]等价a[(1,2)].如果下标元组只包含整数的切片,那么得到的数组和原始数组共享数据,改变得到的数组就会改变原始数组的数据。

>>> x
array([[ 0,  1,  2,  3,  4,  5],
       [ 6,  7,  8,  9, 10, 11],
       [12, 13, 14, 15, 16, 17],
       [18, 19, 20, 21, 22, 23]])
>>> x[1,:] = 0
>>> x
array([[ 0,  1,  2,  3,  4,  5],
       [ 0,  0,  0,  0,  0,  0],
       [12, 13, 14, 15, 16, 17],
       [18, 19, 20, 21, 22, 23]])

数组的下标是元组,所以可以将元组保留下来。例如取出下标为(1,2),(2,4,(3,4)的元素

>>> x
array([[10,  1,  2,  3,  4,  5],
       [ 0,  0,  0,  0,  0,  0],
       [12, 13, 14, 15, 16, 17],
       [18, 19, 20, 21, 22, 23]])
>>> x[(1,2,3),(2,4,4)]
array([ 0, 16, 22])

其中x[(1,2,3),(2,4,4)]得到的元素是x[1,2],x[2,4],x[3,4]

 

以上是关于numpy 多维数组的存取的主要内容,如果未能解决你的问题,请参考以下文章

NumPy学习心得

Numpy数据存取与函数

利用numpy实现多维数组操作图片

numpy中多维数组的绝对索引

深度学习网络中numpy多维数组的说明

将多维元素附加到 numpy 数组中而不进行整形