Python - 从 np.array 中选择复杂的列子集

Question

我想要一个数组 X，并且我想要选择，例如，第 1、第 5、第 7 以及最后 1000 列的所有（不重叠）。我知道我可以使用

X[:, [0,4,6]]

获取第 1、5、7 列，并使用

X[:, -1000:]

获取最后 1000 列。但是有没有一种快速简便的方法可以使用一条线/对象将它们一起选择？

Answer 1

您可以使用

np.r_

：

X[:, np.r_[[0,4,6],-1000:0]]