感悟:列表一遍代表同一纬度操作,元组是不同维度操作

>>> data = np.arange(40).reshape(2, 4, 5)
>>> data
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]]])
>>> data[(1), (1), (1)]
26
>>> data[(1), (1, 2), (1, 3)]
array([26, 33])
>>> data[:, (1, 2), (1, 3)]
array([[ 6, 13],
       [26, 33]])
>>> data[[1], [1, 2], [1, 3]]
array([26, 33])
>>> data[0]
array([[ 0,  1,  2,  3,  4],
       [ 5,  6,  7,  8,  9],
       [10, 11, 12, 13, 14],
       [15, 16, 17, 18, 19]])
>>> data[(0, 1)]
array([5, 6, 7, 8, 9])

>>> data[[0, 1]]
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]]])

>>> 

猜你喜欢

转载自blog.csdn.net/weixin_43906799/article/details/108266947