如何对列表中的元素进行分组？

Question

我有一个清单。每个列表包含三个元素，将它们称为[number, letter, time]。它看起来像这样：my_list = [[0, 'a', '0:00'],           [1, 'a', '1:00'],           [2, 'b', '2:00'],           [3, 'c', '3:00'],           [4, 'b', '4:00'],           [5, 'a', '5:00'],           [6, 'd', '6:00'],           [7, 'c', '7:00'],           [8, 'b', '8:00'],           [9, 'e', '9:00']]我想格式化此列表element[1]，使得letter每个嵌套列表的 成为一个新的嵌套列表，其中和的element[0]出现为和。numbertimeelement[1]element[2]我想要的输出应该是这样的my list = [['a', [0, 1, 5], ['0:00', '1:00', '5:00']],           ['b', [2, 4, 8], ['2:00', '4:00', '8:00']],           ['c', [3, 7], ['3:00', '7:00']],           ['d', [6], ['6:00']],           ['e', [9], ['9:00']]]到目前为止我已经习惯了collections.Counter()获取频率letter>>> letter_count = Counter([element[1] for element in my_list])>>> letter_countCounter({'a': 3, 'b': 3, 'c': 2, 'd': 1, 'e': 1})我相信我可以使用列表理解来完成其余的工作，使用letter_count.keys()and 条件，但我被困住了。有什么建议么？另外，欢迎提出更好、更具描述性的标题（为了子孙后代）的建议。

Answer 1

out = {}

for a, b, c in my_list:

    out.setdefault(b, []).append((a, c))

out = [[k, [i for i, _ in v], [i for _, i in v]]  for k, v in out.items()]

print(out)

印刷：

[['a', [0, 1, 5], ['0:00', '1:00', '5:00']], 

 ['b', [2, 4, 8], ['2:00', '4:00', '8:00']], 

 ['c', [3, 7], ['3:00', '7:00']], 

 ['d', [6], ['6:00']], 

 ['e', [9], ['9:00']]]

Answer 2

另一种方法是使用itertools.groupby

import itertools

new_list = []

my_list = sorted(my_list, key=lambda row: row[1])

for name, groups in itertools.groupby(my_list, key=lambda row: row[1]):

    groups = list(zip(*groups))

    

    grouped_item = [name, groups[0], groups[2]]

    new_list.append(grouped_item)

    

print(new_list)

[['a', (0, 1, 5), ('0:00', '1:00', '5:00')],

 ['b', (2, 4, 8), ('2:00', '4:00', '8:00')],

 ['c', (3, 7), ('3:00', '7:00')],

 ['d', (6,), ('6:00',)],

 ['e', (9,), ('9:00',)]]

Answer 3

这是一种快速而肮脏的方法（与当前其他答案相同的效率，但更多的代码）

my_list = [[0, 'a', '0:00'],

           [1, 'a', '1:00'],

           [2, 'b', '2:00'],

           [3, 'c', '3:00'],

           [4, 'b', '4:00'],

           [5, 'a', '5:00'],

           [6, 'd', '6:00'],

           [7, 'c', '7:00'],

           [8, 'b', '8:00'],

           [9, 'e', '9:00']]

tmp_dict = {}

for first, second, third in my_list:

    if second not in tmp_dict:

        tmp_dict[second] = [[],[]]

    tmp_dict[second][0].append(first)

    tmp_dict[second][1].append(third)

new_list = []

for key, val in tmp_dict.items():

    new_list.append([key] + val)

print(new_list)