数组定义与冒号

>>> import numpy as np
>>> a = np.arange(120).reshape(2,3,4,5)
>>> 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,  24],
         [ 25,  26,  27,  28,  29],
         [ 30,  31,  32,  33,  34],
         [ 35,  36,  37,  38,  39]],

        [[ 40,  41,  42,  43,  44],
         [ 45,  46,  47,  48,  49],
         [ 50,  51,  52,  53,  54],
         [ 55,  56,  57,  58,  59]]],


       [[[ 60,  61,  62,  63,  64],
         [ 65,  66,  67,  68,  69],
         [ 70,  71,  72,  73,  74],
         [ 75,  76,  77,  78,  79]],

        [[ 80,  81,  82,  83,  84],
         [ 85,  86,  87,  88,  89],
         [ 90,  91,  92,  93,  94],
         [ 95,  96,  97,  98,  99]],

        [[100, 101, 102, 103, 104],
         [105, 106, 107, 108, 109],
         [110, 111, 112, 113, 114],
         [115, 116, 117, 118, 119]]]])
分成2块,再分成3块,再4行5列,最后一行右侧有几个中括号,表示是几维的

显示全部数组:

import numpy as np
np.set_printoptions(threshold=np.inf)

冒号:

>>> b = a[:,:,:,:3]
>>> b
array([[[[  0,   1,   2],
         [  5,   6,   7],
         [ 10,  11,  12],
         [ 15,  16,  17]],

        [[ 20,  21,  22],
         [ 25,  26,  27],
         [ 30,  31,  32],
         [ 35,  36,  37]],

        [[ 40,  41,  42],
         [ 45,  46,  47],
         [ 50,  51,  52],
         [ 55,  56,  57]]],


       [[[ 60,  61,  62],
         [ 65,  66,  67],
         [ 70,  71,  72],
         [ 75,  76,  77]],

        [[ 80,  81,  82],
         [ 85,  86,  87],
         [ 90,  91,  92],
         [ 95,  96,  97]],

        [[100, 101, 102],
         [105, 106, 107],
         [110, 111, 112],
         [115, 116, 117]]]])
>>> c = a[:,:,:,3:3+3]
>>> c
array([[[[  3,   4],
         [  8,   9],
         [ 13,  14],
         [ 18,  19]],

        [[ 23,  24],
         [ 28,  29],
         [ 33,  34],
         [ 38,  39]],

        [[ 43,  44],
         [ 48,  49],
         [ 53,  54],
         [ 58,  59]]],


       [[[ 63,  64],
         [ 68,  69],
         [ 73,  74],
         [ 78,  79]],

        [[ 83,  84],
         [ 88,  89],
         [ 93,  94],
         [ 98,  99]],

        [[103, 104],
         [108, 109],
         [113, 114],
         [118, 119]]]])

猜你喜欢

转载自blog.csdn.net/infinita_LV/article/details/86159519