Numpy选择特定行列

Posted 2022-08-27 算法与编程之美

问题

方法

import numpy as np

a = np.random.randint(0, 100, (3,3))
print(a)

b = a[:, [0,2]] # 选择所有行, 第1、3列
print(b)

结语