çıkışı iç içe liste

Question

Aşağıdaki liste var:

a = [[0,11], [1,12], [3,14], [5,16],[7,18]] 

Ben

a[:,0] > 3 

satistying tüm unsurları alır yeni bir liste b oluşturmak ve a yılında endeksi başa eklemek istiyorum seçili yuvalanmış liste.

b = [[3, 5, 16],[4, 7,18]] 

ile

: endeksi ile

b = [points, points in a if points[0]>3] 

irade çıkışı b = [[5, 16],[7,18]] ve

b = [[index, points], index, points in enumerate(a) if points[0]>3] 

bir hata gösteriyor Yani b shoudl benziyor. Liste kavramasını kullanarak b'u nasıl oluşturabilirim?

Sen iç içe listeleri öğeleri tutmak için ıskarta değişkenlerin bir demet kullanabilirsiniz

Answer 1

:

>>> [[i,j,k] for i,(j,k) in enumerate(a) if j>3] 
[[3, 5, 16], [4, 7, 18]] 

Veya daha fazla öğe içeren listelere için daha kapsamlı bir yaklaşım olarak aşağıdaki liste anlama kullanabilirsiniz:

>>> [[i] + points for i, points in enumerate(a) if points[0]>3] 
[[3, 5, 16], [4, 7, 18]] 

Python 3 versiyonu kullanılarak açma atama: Eh

>>> [[i, first] + rest for i,(first, *rest) in enumerate(a) if first>3] 
[[3, 5, 16], [4, 7, 18]] 

Answer 2

map(lambda y: [y[0]] + y[1], filter(lambda x: x[1][0] > 3, enumerate(a)))

Answer 3

a Eğer (öyle görünüyor ki bir NumPy dizidir: Ben işte o (oldukça şifreli hala liste comprehensions olmadan ama) yapmak için başka bir yol, bu soruyu ama benden önce yayınlanan diğer insanları yanıtlayan başladı

# One of the rare cases when numpy.where isn't superfluous. 
indices = numpy.where(a[:, 0] > 3)[0] 

index_column = indices[:, None] 
selected_rows = a[indices] 

b = numpy.hstack([index_column, selected_rows]) 

Veya daha az ara değişkenlerle:

indices = numpy.where(a[:, 0] > 3)[0] 
b = numpy.hstack([indices[:, None], a[indices]]) 

) gerçek kullanım durumunda olması, aşağıdakileri yapabileceğini

Büyük boyutlarda a, bu muhtemelen enumerate veya diğer Python düzeyinde yineleme tekniklerine dayanan çözümleri daha iyi performans gösterecektir.