Python中numpy读取高维数组中特定的行，列

直接跑代码，看懂代码，跑一遍，你自然懂了。

'''
多维矩阵按括号的层级，从外向内，一次是第1，2，3，...维

b[]内用逗号将各维分开，分别代表第1,2,3...维元素
逗号把数组分成两维度，左边是所有的行（第一维度），右边是所有的列（第二维度元素）
每个维度上都有自己的下标，也可以用':'取部分
'''
import numpy as np

a=np.arange(10)
print(a)
print(a[0:9])  # 包头不包尾
print(a[3:6])
print(a[:5])  # :前面不写就是从下标为0开始
print(a[5:])  # :后面不写就是一直到最后一个元素
print(a[:])   # :前后都不写就是从头到尾



print('---------------')

b= np.mat(np.arange(20).reshape(4,5))  #建立4*5的二维矩阵，数字从0到20之间
print(b)
print(b[1:3,2:5])   # 先取第一维中下标为1,2的2部分，再取第二维中下标为2,3,4的3部分；逗号左边，1,2行，逗号右边，2,3,4,列，他们交集就是我们要的
print(b[:2,2:])     # 同理，前面不写从头开始，后面不写一直到末尾，逗号左边：0,1两行，逗号右边，2,3,4列，他们交集就是我们要的
print(b[:2,3])      # 当然，也可以在某维度上只取一行
print(b[:,4])    #，逗号左边，所有行，逗号右边，第四列
print('-----------------')

c= np.arange(60).reshape(3,4,5)
print(c)
print('-------c[:2,2:4,1:4]------------')

print(c[:2,2:4,1:4])  # 从外向内一层一层的割，切割不改变矩阵维度
print('---------c[:,:,4]----------')

print(c[:,:,4])
print('-------d-----------')
d= np.arange(240).reshape(3,4,5,4)

print(d)
print('-------d[:2,1:3,2:5,1:3]------------')

print(d[:2,1:3,2:5,1:3])
print('-------d[:,:,:,3]------------')
print(d[:,:,:,3])