简而言之,您可以：
A[:, range(3) + [6, 7]]
要么
A[:, np.r_[:3, 6, 7]]
要理解为什么你的第一次尝试不起作用,你需要更多地了解一般的python索引以及numpy的索引是如何工作的.
首先,切片表示法仅在切片内有效,因此blah = [：3,6,7]是无效的python语法,因为在这种情况下你正在构建一个新的列表. (你隐含地使用A [：,[：3,6,7]]来做这个.第二组括号创建一个新列表.)Numpy有一个特殊的(但经常讨厌)运算符来使用切片表示法来构造类似于你想要的东西：np.r_.例如：
In [1]: print np.r_[:3, 6, 7]
[0 1 2 6 7]请注意,通过将两个列表一起添加,我们可以得到或多或少相同的结果(np.r_返回一个数组,下一个示例将返回一个列表)：
In [2]: print range(3) + [6, 7]
[0, 1, 2, 6, 7]然而,了解numpy窗帘背后发生的事情也很重要. numpy中有两种常见的索引类型. “正常”索引使用任何排序的切片并返回数组的视图.数据不会被复制. “花式”索引使用任意项目序列(例如列表)并复制数据.
因为可以通过标准切片表示法描述的任何内容都有常规步骤(即开始,结束和步骤间隔),所以您可以创建新数组而无需复制原始数据. (Numpy数组必须在内存中“定期跨越”.你可以在不复制数据的情况下引用“每三个项目”,但不能引用“第2,5和6项”,因为后者没有常规模式.)
所有这一切可能看起来令人困惑,但这里有一个重要原因的例子.让我们做一个示例数组并将它切成两种不同(但等效)的方式：
In [1]: a = np.arange(18).reshape(3, 6)
In [2]: a
Out[2]:
array([[ 0, 1, 2, 3, 4, 5],
[ 6, 7, 8, 9, 10, 11],
[12, 13, 14, 15, 16, 17]])
In [3]: b = a[:, :3]
In [4]: b
Out[4]:
array([[ 0, 1, 2],
[ 6, 7, 8],
[12, 13, 14]])
In [5]: c = a[:, [0, 1, 2]]
In [6]: c
Out[6]:
array([[ 0, 1, 2],
[ 6, 7, 8],
[12, 13, 14]])b和c看起来完全相同.但是,c是一段时间内数据的新副本,b引用原始数据.如果我们更改c,则不会修改a：
In [7]: c[0, 0] = 10000
In [8]: c
Out[8]:
array([[10000, 1, 2],
[ 6, 7, 8],
[ 12, 13, 14]])
In [9]: a
Out[9]:
array([[ 0, 1, 2, 3, 4, 5],
[ 6, 7, 8, 9, 10, 11],
[12, 13, 14, 15, 16, 17]])如果我们改变b,a将被修改：
In [10]: a
Out[10]:
array([[ 0, 1, 2, 3, 4, 5],
[ 6, 7, 8, 9, 10, 11],
[12, 13, 14, 15, 16, 17]])
In [11]: b[0,0] = 99999
In [12]: a
Out[12]:
array([[99999, 1, 2, 3, 4, 5],
[ 6, 7, 8, 9, 10, 11],
[ 12, 13, 14, 15, 16, 17]])
In [13]: b
Out[13]:
array([[99999, 1, 2],
[ 6, 7, 8],
[ 12, 13, 14]])这使您可以对内存使用进行大量控制,并且允许numpy非常高效(当您开始在内存中使用非常大的数组时,这非常重要.).但是,如果您不知道发生了什么,您可能会被它灼伤.
