2013-03-01 10 views
9

Mam dwie listy Lx i Ly, każdy element z Lx ma odpowiednią etykietę w Ly. Przykład:wybieranie elementów z listy zgodnie z ich etykietami z drugiej listy

Lx = [[1,2,5], [5,2,7], [7,0,4], [9,2,0], [1,8,5], [3,4,5], [3,2,7], [2,9,7]] 
Ly = [A, C, A, B, A, B, C, C] 

Jak łatwo mogę uzyskać listę/etykietę, gdzie elementy listy są elementami z Lx, które mają tę samą etykietę w Ly? Czyli:

[[1,2,5], [7,0,4], [1,8,5]] 
[[5,2,7], [3,2,7], [2,9,7]] 
[[9,2,0], [3,4,5]] 

Odpowiedz

7
Lx = [[1,2,5], [5,2,7], [7,0,4], [9,2,0], [1,8,5], [3,4,5], [3,2,7], [2,9,7]] 
Ly = ['A', 'C', 'A', 'B', 'A', 'B', 'C', 'C'] 
d = {} 
for x,y in zip(Lx,Ly): 
    d.setdefault(y, []).append(x) 

d jest teraz:

{'A': [[1, 2, 5], [7, 0, 4], [1, 8, 5]], 
'B': [[9, 2, 0], [3, 4, 5]], 
'C': [[5, 2, 7], [3, 2, 7], [2, 9, 7]]} 
+0

@shn Jeśli wykonujesz '' z kolekcji import OrderedDict'' i '' d = OrderedDict() '', kolejność list w '' d.values ​​() '' odzwierciedla kolejność elementów w Ly – eyquem

1

Poniższa będzie Ci dość blisko:

from collections import defaultdict 

Lx = [[1,2,5], [5,2,7], [7,0,4], [9,2,0], [1,8,5], [3,4,5], [3,2,7], [2,9,7]] 
Ly = ['A', 'C', 'A', 'B', 'A', 'B', 'C', 'C'] 

d = defaultdict(list) 
for x, y in zip(Lx, Ly): 
    d[y].append(x) 
d = dict(d) 

print(d) 

ta produkuje

{'A': [[1, 2, 5], [7, 0, 4], [1, 8, 5]], 'C': [[5, 2, 7], [3, 2, 7], [2, 9, 7]], 'B': [[9, 2, 0], [3, 4, 5]]}