美文网首页
多维数组切片

多维数组切片

作者: 猴子喜 | 来源:发表于2018-12-05 15:00 被阅读0次
>>> a = np.arange(24).reshape(2,3,4)
>>> a
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]]])

我们可以形象地把它看做一个两层楼建筑,每层楼有12个房间,并排列成3行4列。

  1. 选定任意一个房间,即楼层、行号和列号。例如:选定第1层楼、第1行、第1列的房间,可以这样表示:
    >>> a[0, 0, 0] 0

  2. 不关心楼层,也就是说要选取所有楼层的第1行、第1列的房间
    >>> a[:,0,0] [ 0 12]

  3. 选取所有楼层的第1行、第1列的房间
    >>> a[:, 0, 0] array([ 0, 12])

多个冒号可以用一个省略号(…)来代替
>>> a[0, ...] array([[ 0, 1, 2, 3], [ 4, 5, 6, 7], [ 8, 9, 10, 11]])

参考

https://blog.csdn.net/qiqicos/article/details/78971406

相关文章

网友评论

      本文标题:多维数组切片

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