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

我有一个清单。每个列表包含三个元素,将它们称为[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_count

Counter({'a': 3, 'b': 3, 'c': 2, 'd': 1, 'e': 1})

我相信我可以使用列表理解来完成其余的工作,使用letter_count.keys()and 条件,但我被困住了。有什么建议么?


另外,欢迎提出更好、更具描述性的标题(为了子孙后代)的建议。


弑天下
浏览 114回答 3
3回答

慕田峪9158850

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

慕神8447489

另一种方法是使用itertools.groupbyimport itertoolsnew_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',)]]

一只甜甜圈

这是一种快速而肮脏的方法(与当前其他答案相同的效率,但更多的代码)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)
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Python