版权声明:知识无界限,大家可自由转载。 https://blog.csdn.net/xidianbaby/article/details/89373680
import numpy as np
data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(data.shape) # (3, 3)
sel_data = data[:, [2, 1]] # 原数据的第三列在此数组的第一列
print(sel_data.shape) # (3, 2)