Poniższy kod Pythona wydaje się być bardzo rozwlekły, gdy pochodzi z tła Matlab
>>> a = [1, 2, 3, 1, 2, 3]
>>> [index for index,value in enumerate(a) if value > 2]
[2, 5]
Będąc w Matlabie mogę napisać:
>> a = [1, 2, 3, 1, 2, 3];
>> find(a>2)
ans =
3 6
Czy istnieje metoda krótkiej ręki na napisanie tego w Pythonie, czy po prostu trzymam się długiej wersji?
Dziękuję za wszystkie sugestie i wyjaśnienie uzasadnienia składni Pythona.
Po znalezieniu następujących informacji na stronie numpy myślę, że znalazłem rozwiązanie, które mi się podoba:
http://docs.scipy.org/doc/numpy/user/basics.indexing.html#boolean-or-mask-index-arrays
Zastosowanie informacji z tej witryny do mojego problemu powyżej dałoby:
>>> 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]
Poniższe powinno wtedy działać (ale nie mam pod ręką interpretera Pythona, aby go przetestować):
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]
[idx for idx in range(len(a)) if a[idx] > 2]
? Powodem, dla którego jest to trochę niewygodne w Pythonie, jest to, że nie używa on indeksów tak często, jak inne języki.