Подтвердить что ты не робот

Обозначение фрагмента Python с запятой/списком

Я столкнулся с некоторым кодом python с записью среза, с которой мне сложно разобраться. Это похоже на нотацию фрагмента, но использует запятую и список:

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

Является ли этот синтаксис действительным? Если да, то что он делает?

edit выглядит как массив 2D numpy

4b9b3361

Ответ 1

Предполагая, что объект действительно является массивом numpy, это называется расширенная индексация и выделяет указанные столбцы:

>>> 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])

То, что оно фактически выполняет, зависит от индексируемого типа.