División de listas de Python con índices arbitrarios

Question

¿Existe alguna forma mejor de extraer índices arbitrarios de una lista en python?

El método que utilizo actualmente es:

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

donde A es la matriz quiero cortar, y [5,13,25] son ​​los elementos que quiero conseguir. Parece mucho más detallado que el equivalente de 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

No hay manera "ya hecho" - la forma en que lo hace es bastante ingenua, y usted podría utilizarlo. Si tiene mucho de eso a través de su código, es posible que desee utilizar una subclase de lista que usaría una sintaxis como matlabs; se puede hacer en un código de algunas líneas, la mayor carga es que tendría que el trabajo siempre usa esta nueva clase en lugar de las listas incorporadas.

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

Y en la consola:

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

Answer 3

Parece igual que lo haría:

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

esto parece ser casi idéntica a la versión de MATLAB.

Answer 4

Si usted es un usuario de Matlab, pero desea utilizar Python, echa un vistazo a 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]) 

Aquí es una comparison of NumPy and Matlab, y aquí es una tabla de Matlab commands and their equivalents in NumPy común.