2021-09-25

python中的数组切分

数组切分，a[:, [-1]]，a[:-1, :-1]

#先定义一个数组
a = np.array([[1, 2, 3, 4, 5, 6],
              [5, 6, 7, 8, 9, 0],
              [2, 3, 4, 5, 6, 7],
              [3, 4, 1, 2, 7, 6]])

#1:3表示取第2，3两行, 2:4表示取第2，3，4三列
print(a[1:3, 1:4])  
'''
result:
[[6 7 8]
 [3 4 5]]
'''

 #取所有行，取第2，3，4三列
print(a[:, 1:4])   
'''
result:
[[2 3 4]
 [6 7 8]
 [3 4 5]
 [4 1 2]]
'''

 #取除最后一行和最后一列的所有行，列
print(a[:-1, :-1]) 
'''
result:
[[1 2 3 4 5]
 [5 6 7 8 9]
 [2 3 4 5 6]]
'''

#取所有行，取除最后一列所有列
print(a[:, :-1]) 
'''
result:
[[1 2 3 4 5]
 [5 6 7 8 9]
 [2 3 4 5 6]
 [3 4 1 2 7]]
'''

#取所有行中最后一列，与上一步是互补的
print(a[:, [-1]])  
'''
result:
[[6]
 [0]
 [7]
 [6]]
'''