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])

猜你喜欢

转载自blog.csdn.net/BTUJACK/article/details/82782575
今日推荐