Python numpy 多维数组切片

import numpy as np

np.random.seed(1234)
A = np.random.randint(0, 6,(5,3))  #生成10行3列数值在[0,6）的随机数组
print(A)
'''结果
[[3 5 4]
 [4 0 1]
 [1 1 2]
 [3 4 4]
 [2 2 0]]
'''

print(A[0:2 , 0:3])  #第0行和第1行 的 第0,1,2列
# 逗号是分割点， 前面的0:2指的是行，后面的0:3是列
'''
[[3 5 4]
 [4 0 1]]
'''
print(A[1:3 , 0:2])
'''结果
[[4 0]
 [1 1]]
'''

print(A[3 , :])  #[3 4 4]
print(A[3])      #[3 4 4]

print(A[:3 , :])
'''
[[3 5 4]
 [4 0 1]
 [1 1 2]]
'''

print(A[: , 1])    #所有行，中间那列（从0开始算，第1列），注意返回的是一维的数组
'''
[5 0 1 4 2]
'''

print(A[: , 1:2])  #所有行， [1,2)那列（从0开始算，也就是中间那列）  这里就是一个二维的数组
'''
[[5]
 [0]
 [1]
 [4]
 [2]]
'''

参考：python 多维切片_stashbllog-CSDN博客_python 多维切片

Python多维数组切片 - Abraham_Xu - 博客园 (cnblogs.com)