Python liste anlama özelliğini kullanarak bir koşula göre elemanların dizinini bulma

Aşağıdaki Python kodu, bir Matlab arka planından geldiğinde çok uzun soluklu görünüyor

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

Matlab'dayken şunları yazabilirim:

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

Bunu Python'da yazmanın kısa bir yöntemi var mı, yoksa sadece uzun versiyona mı bağlıyım?

Python'un sözdiziminin gerekçesinin tüm önerileri ve açıklamaları için teşekkür ederiz.

Numpy web sitesinde aşağıdakileri bulduktan sonra, beğendiğim bir çözüm bulduğumu düşünüyorum:

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

Bu web sitesindeki bilgileri yukarıdaki sorunuma uygulamak aşağıdakileri verir:

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

Aşağıdakiler daha sonra çalışmalıdır (ancak test etmek için elimde bir Python yorumlayıcısı yok):

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]

• Python'da bunun için indeks kullanmazsınız, sadece değerlerle ilgilenirsiniz— [value for value in a if value > 2]. Genellikle dizinlerle uğraşmak, bir şeyi en iyi şekilde yapmadığınız anlamına gelir.

• Eğer varsa do Matlab'ınkine benzer bir API ihtiyaç şunları kullanırsınız numpy , çok boyutlu diziler ve ağır Matlab esinlenmiştir Python sayısal matematik için bir paket. Liste yerine uyuşmuş bir dizi kullanırsınız.

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

Diğer yol:

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

Genel olarak, hazır bir işlev olmasına rağmen find, liste anlamalarının genel ve dolayısıyla çok güçlü bir çözüm olduğunu unutmayın . findPython'da bir işlev yazmanızı ve daha sonra istediğiniz gibi kullanmanızı hiçbir şey engellemez . yani:

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

Matlab'ı taklit etmek için buradaki listeleri kullandığımı unutmayın. Üreteçleri ve yineleyicileri kullanmak daha Pythonic olurdu.

Benim için iyi çalışıyor:

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

Belki başka bir soru şudur: "Bu endeksleri aldıktan sonra ne yapacaksın?" Bunları başka bir liste oluşturmak için kullanacaksanız, Python'da bunlar gereksiz bir orta adımdır. Belirli bir koşulla eşleşen tüm değerleri istiyorsanız, yerleşik filtreyi kullanın:

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

Veya kendi listenizi yazın:

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

Bunları listeden çıkarmak istiyorsanız, Pythonic yolu listeden illa ki çıkarılmak zorunda değildir, ancak yeni bir liste oluşturuyormuşsunuz gibi bir liste anlayışı yazmak listvar[:]ve sol taraftaki -yan:

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

Matlab find, dizi merkezli modeli, dizi indekslerini kullanarak öğeleri seçerek çalıştığı için sağlar. Bunu kesinlikle Python'da yapabilirsiniz , ancak daha Pythonic yolu @EliBendersky tarafından daha önce bahsedildiği gibi yineleyiciler ve üreteçler kullanmaktır.

Geç cevap olsa bile: Sanırım bu hala çok iyi bir soru ve IMHO Python (numpy gibi ek kitaplıklar veya araç setleri olmadan) hala manuel olarak tanımlanmış bir filtreye göre liste öğelerinin dizinlerine erişmek için uygun bir yöntemden yoksun.

Bu işlevi sağlayan bir işlevi manuel olarak tanımlayabilirsiniz:

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

Verim: [0, 4]

Benim hayal gücüme göre mükemmel bir yol, bir alt sınıf liste yapmak ve indeksleri eklemek sınıf yöntemi olarak işlev görmek olurdu. Bu şekilde yalnızca filtre yöntemine ihtiyaç duyulur:

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)

Bu konuda biraz daha detaylandırdım: http://tinyurl.com/jajrr87