Jak grupować podobne elementy na liście?

Question

Szukam grupowania podobnych elementów na liście na podstawie pierwszych trzech znaków w ciągu. Na przykład:

test = ['abc_1_2', 'abc_2_2', 'hij_1_1', 'xyz_1_2', 'xyz_2_2'] 

Jak mogę grupa powyższe elementy listy na grupy na podstawie pierwszej z grup liter (np 'abc')? Poniżej znajduje się przeznaczony wyjściowa:

output = {1: ('abc_1_2', 'abc_2_2'), 2: ('hij_1_1',), 3: ('xyz_1_2', 'xyz_2_2')} 

lub

output = [['abc_1_2', 'abc_2_2'], ['hij_1_1'], ['xyz_1_2', 'xyz_2_2']] 

---

Próbowałem, używając itertools.groupby do osiągnięcia tego celu bez powodzenia:

>>> import os, itertools 
>>> test = ['abc_1_2', 'abc_2_2', 'hij_1_1', 'xyz_1_2', 'xyz_2_2'] 
>>> [list(g) for k.split("_")[0], g in itertools.groupby(test)] 
[['abc_1_2'], ['abc_2_2'], ['hij_1_1'], ['xyz_1_2'], ['xyz_2_2']] 

---

Mam spojrzał na następujące posty bez powodzenia:

How to merge similar items in a list. Przykład grupuje podobne elementy (np. 'house' i 'Hose'), stosując podejście, które jest zbyt skomplikowane dla mojego przykładu.

How can I group equivalent items together in a Python list?. Tam właśnie znalazłem pomysł na zrozumienie listy.

Answer 1

Część .split("_")[0] powinna znajdować się wewnątrz funkcji jednoargumentowej, którą podaje się jako drugi argument do itertools.groupby.

>>> import os, itertools 
>>> test = ['abc_1_2', 'abc_2_2', 'hij_1_1', 'xyz_1_2', 'xyz_2_2'] 
>>> [list(g) for _, g in itertools.groupby(test, lambda x: x.split('_')[0])] 
[['abc_1_2', 'abc_2_2'], ['hij_1_1'], ['xyz_1_2', 'xyz_2_2']] 
>>> 

Mając go w for ... części nic nie robi, ponieważ wynik jest natychmiast odrzucane.

---

Również byłoby nieco bardziej wydajny w użyciu str.partition kiedy chcesz tylko jeden podział:

[list(g) for _, g in itertools.groupby(test, lambda x: x.partition('_')[0])] 

Demo:

>>> from timeit import timeit 
>>> timeit("'hij_1_1'.split('_')") 
1.3149855638076913 
>>> timeit("'hij_1_1'.partition('_')") 
0.7576401470019234 
>>> 

To nie jest głównym problemem, jak obie metody są dość szybkie na małych strunach, ale pomyślałem, że wspomnę o tym.