Нарезка списков Python с произвольными индексами

Question

Есть ли лучший способ извлечь произвольные индексы из списка в python?

Метод, который я в настоящее время использую:

a = range(100) 
s = [a[i] for i in [5,13,25]] 

Если это массив, который я хочу, чтобы нарезать и [5,13,25] являются элементы, которые я хочу получить. Это кажется намного более многословен, чем эквивалент Matlab:

a = 0:99; 
s = a([6,14,26]) 

Answer 1

>>> from operator import itemgetter 
>>> a = range(100) 
>>> itemgetter(5,13,25)(a) 
(5, 13, 25) 

Answer 2

Там нет «готовых» путь - путь вы делаете это довольно бесхитростный, и вы можете использовать его. Если у вас есть много этого кода, вы можете использовать подкласс списка, который будет использовать синтаксис так же, как matlabs - это можно сделать в нескольких строках кода, основное бремя - это то, что вам придется работа всегда использует этот новый класс вместо встроенных списков.

class MyList(list): 
    def __getitem__(self, index): 
     if not isinstance(index, tuple): 
      return list.__getitem__(self, index) 
     return [self[i] for i in index] 

А на консоли:

>>> m = MyList(i * 3 for i in range(100)) 
>>> m[20, 25,60] 
[60, 75, 180] 

Answer 3

Похоже, что вы могли бы сделать:

a = list(range(99)) 
    s = [a[5], a[13], a[25]] 

это, кажется, почти идентична версии Matlab.

Answer 4

Если вы являетесь пользователем Matlab, но хотите использовать Python, проверить numpy:

In [37]: import numpy as np 

In [38]: a = np.arange(100) 

In [39]: s = a[[5,13,25]] 

In [40]: s 
Out[40]: array([ 5, 13, 25]) 

Вот comparison of NumPy and Matlab, а вот таблица общей Matlab commands and their equivalents in NumPy.