numpy 矩阵提取某几行

import numpy as np
a = np.array([1,2,3,4,5,6,7,8,9,10,11,12])
a = a.reshape([4,3])

print('a: ')
print(a)

 # 提取a 的第1行
print('\n a[0,:]=')
print(a[0,:])  

# 提取a的第2，3行
print('\n a[1:3, :]=')
print(a[1:3, :])   

 # 提取a的第1行和第3行
print('\n a[[0,2], :]=')
print(a[[0,2],:]) 


b = np.array([0, 2])
b = b.astype(np.int32)
# 提取a的第1行和第3行
print('\n a[b,]=')
print(a[ b, ])  

# 提取a的第1行
print('\n a[b[0],:]=')
print(a[ b[0], ])   

列的提取方式类似，此处就不列举了。

运行结果图如下：