如何用一个命令提取numpy数组的切片和特定列？

Question

我想在一个命令中提取numpy数组的列片和特定列。

例如，对于数组A：

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

我想从0到2和列4一起选择列。解决方案A[:,(0,1,2,4)]在这个例子中很容易实现。对于较大的数组，我需要一个A[:,(0:3,4)]类型的命令来选择切片和特定列。命令A[:,(0:3,4)]不工作。

是否有一种实用而优雅的方法可以用一个命令提取切片和特定列？

事先非常感谢！

Answer 1

正如ombk所建议的，您可以使用_r__。它是连接切片表达式的完美工具。

就你而言：

A[:, np.r_[0:3, 4]]

检索数组的预期部分。

同样，您也可以连接更多的切片表达式。