Encontrar el índice de elementos basado en una condición usando la comprensión de la lista de Python

El siguiente código de Python parece ser muy largo cuando proviene de un fondo de Matlab

>>> a = [1, 2, 3, 1, 2, 3]
>>> [index for index,value in enumerate(a) if value > 2]
[2, 5]

Cuando en Matlab puedo escribir:

>> a = [1, 2, 3, 1, 2, 3];
>> find(a>2)
ans =
     3     6

¿Existe un método de mano corta para escribir esto en Python, o simplemente me quedo con la versión larga?

Gracias por todas las sugerencias y explicaciones del fundamento de la sintaxis de Python.

Después de encontrar lo siguiente en el sitio web de numpy, creo que encontré una solución que me gusta:

http://docs.scipy.org/doc/numpy/user/basics.indexing.html#boolean-or-mask-index-arrays

Aplicar la información de ese sitio web a mi problema anterior, daría lo siguiente:

>>> from numpy import array
>>> a = array([1, 2, 3, 1, 2, 3])
>>> b = a>2 
array([False, False, True, False, False, True], dtype=bool)
>>> r = array(range(len(b)))
>>> r(b)
[2, 5]

Lo siguiente debería funcionar (pero no tengo un intérprete de Python a mano para probarlo):

class my_array(numpy.array):
    def find(self, b):
        r = array(range(len(b)))
        return r(b)


>>> a = my_array([1, 2, 3, 1, 2, 3])
>>> a.find(a>2)
[2, 5]

• En Python, no usarías índices para esto en absoluto, solo lidiarías con los valores— [value for value in a if value > 2]. Por lo general, tratar con índices significa que no está haciendo algo de la mejor manera.

• Si usted no necesita una API similar a Matlab, se usaría numpy , un paquete de matrices multidimensionales y matemáticas numérica en Python que está fuertemente inspirado en Matlab. Estaría usando una matriz numpy en lugar de una lista.

  >>> import numpy
  >>> a = numpy.array([1, 2, 3, 1, 2, 3])
  >>> a
  array([1, 2, 3, 1, 2, 3])
  >>> numpy.where(a > 2)
  (array([2, 5]),)
  >>> a > 2
  array([False, False,  True, False, False,  True], dtype=bool)
  >>> a[numpy.where(a > 2)]
  array([3, 3])
  >>> a[a > 2]
  array([3, 3])

De otra manera:

>>> [i for i in range(len(a)) if a[i] > 2]
[2, 5]

En general, recuerde que si bien findes una función lista para usar, las listas por comprensión son una solución general y, por lo tanto, muy poderosa . Nada le impide escribir una findfunción en Python y usarla más tarde como desee. Es decir:

>>> def find_indices(lst, condition):
...   return [i for i, elem in enumerate(lst) if condition(elem)]
... 
>>> find_indices(a, lambda e: e > 2)
[2, 5]

Tenga en cuenta que estoy usando listas aquí para imitar Matlab. Sería más Pythonic usar generadores e iteradores.

Para mi funciona bien:

>>> import numpy as np
>>> a = np.array([1, 2, 3, 1, 2, 3])
>>> np.where(a > 2)[0]
[2 5]

Quizás otra pregunta sea, "¿qué vas a hacer con esos índices una vez que los obtengas?" Si los va a usar para crear otra lista, en Python, son un paso intermedio innecesario. Si desea que todos los valores coincidan con una condición determinada, simplemente use el filtro incorporado:

matchingVals = filter(lambda x : x>2, a)

O escriba su propia lista de comprensión:

matchingVals = [x for x in a if x > 2]

Si desea eliminarlos de la lista, entonces la forma Pythonic no es necesariamente eliminarlos de la lista, sino escribir una lista de comprensión como si estuviera creando una nueva lista y volviendo a asignar en el lugar usando el listvar[:]de la izquierda. -lado:

a[:] = [x for x in a if x <= 2]

Matlab suministra findporque su modelo centrado en matrices funciona seleccionando elementos utilizando sus índices de matriz. Usted puede hacer esto en Python, sin duda, pero la forma más Pythonic está utilizando iteradores y generadores, como ya se ha mencionado por @EliBendersky.

Incluso si es una respuesta tardía: creo que esta sigue siendo una muy buena pregunta y, en mi humilde opinión, Python (sin bibliotecas o kits de herramientas adicionales como numpy) todavía carece de un método conveniente para acceder a los índices de los elementos de la lista de acuerdo con un filtro definido manualmente.

Puede definir manualmente una función, que proporciona esa funcionalidad:

def indices(list, filtr=lambda x: bool(x)):
    return [i for i,x in enumerate(list) if filtr(x)]

print(indices([1,0,3,5,1], lambda x: x==1))

Rendimientos: [0, 4]

En mi imaginación, la manera perfecta sería hacer una lista de clase secundaria y agregar la función de índices como método de clase. De esta manera solo se necesitaría el método de filtro:

class MyList(list):
    def __init__(self, *args):
        list.__init__(self, *args)
    def indices(self, filtr=lambda x: bool(x)):
        return [i for i,x in enumerate(self) if filtr(x)]

my_list = MyList([1,0,3,5,1])
my_list.indices(lambda x: x==1)

Elaboré un poco más sobre ese tema aquí: http://tinyurl.com/jajrr87