带有逗号/列表的 Python 切片表示法答案

【问题标题】：Python Slice Notation with Comma/List带有逗号/列表的 Python 切片表示法
【发布时间】：2013-06-26 14:06:30
【问题描述】：

我遇到了一些带有切片表示法的 python 代码，但我无法弄清楚。它看起来像切片符号，但使用逗号和列表：

list[:, [1, 2, 3]]

这种语法有效吗？如果有，它有什么作用？

edit 看起来像是一个 2D numpy 数组

【问题讨论】：

您确定list 是一个列表吗？它可能是一个以不寻常的方式覆盖__getitem__ 的类的实例。
你试过type(list)吗？可以是 numpy 数组或 pandas 数据框。
@mattexx，我没有完整的代码，这在 numpy 和 pandas 中有什么作用？
Numpy 数组和 pandas 数据帧使用二维数组，很像 matlab 和 r。第一个参数通常代表一个行切片，而第二个参数代表一个列切片。所以 list[:,:3] 将是所有行，前三列。我不确定是否支持您在问题中使用的确切符号。
@mattexx：为了不让 OP 感到困惑，强调 numpy 数组与其他一些数据结构的本质上不同，它可能不是二维对象。他们是 n-D。

标签： python list numpy slice

【解决方案1】：

假设对象真的是一个numpy数组，这被称为advanced indexing，并挑选出指定的列：

>>> import numpy as np
>>> a = np.arange(12).reshape(3,4)
>>> a
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])
>>> a[:, [1,2,3]]
array([[ 1,  2,  3],
       [ 5,  6,  7],
       [ 9, 10, 11]])
>>> a[:, [1,3]]
array([[ 1,  3],
       [ 5,  7],
       [ 9, 11]])

请注意，这不适用于标准 Python 列表：

>>> a.tolist()
[[0, 1, 2, 3], [4, 5, 6, 7], [8, 9, 10, 11]]
>>> a.tolist()[:,[1,2,3]]
Traceback (most recent call last):
  File "<ipython-input-17-7d77de02047a>", line 1, in <module>
    a.tolist()[:,[1,2,3]]
TypeError: list indices must be integers, not tuple

【讨论】：

【解决方案2】：

它生成一个复数并将其传递给__*item__()：

>>> class Foo(object):
...   def __getitem__(self, val):
...     print val
... 
>>> Foo()[:, [1, 2, 3]]
(slice(None, None, None), [1, 2, 3])

它实际上执行什么取决于被索引的类型。

【讨论】：