在与整数配对的所有数字列表中的元组内创建一个列表[重复]

Posted 2023-02-23

【中文标题】在与整数配对的所有数字列表中的元组内创建一个列表[重复]

【英文标题】：Create a list inside a tuple inside a list of all numbers paired with an integer [duplicate]

【发布时间】：2018-05-03 05:29:23

【问题描述】：

我有一个这样的列表：

[(0, 1), (0, 2), (0, 3), (1, 4), (1, 6), (1, 7), (1, 9)]

我需要转换成如下所示的元组：

[(0, [1, 2, 3]), (1, [4, 6, 7, 9])]

这是我的代码：

friends = open(file_name).read().splitlines()
network = []
friends = [tuple(int(y) for y in x.split(' ')) for x in friends]
return friends

【问题讨论】：

使用ast.literal_eval从文件中解析列表。

【参考方案1】：

你可以试试这个：

import itertools
s = [(0, 1), (0, 2), (0, 3), (1, 4), (1, 6), (1, 7), (1, 9)]
final_data = [(a, [i[-1] for i in list(b)]) for a, b in itertools.groupby(sorted(s, key=lambda x:x[0]), key=lambda x:x[0])]

输出：

[(0, [1, 2, 3]), (1, [4, 6, 7, 9])]

【参考方案2】：

这不是您所要求的，但从您正在寻找的输出类型来看，我建议您使用defaultdict，它具有超级可读性和高效性，

from collections import defaultdict

some_list = [(0, 1), (0, 2), (0, 3), (1, 4), (1, 6), (1, 7), (1, 9)]

d = defaultdict(list)

for k, v in some_list:
    d[k].append(v)

输出

defaultdict(<class 'list'>, 0: [1, 2, 3], 1: [4, 6, 7, 9])

【参考方案3】：

“单行”嵌套列表推导式，带有换行符

tpls = [(0, 1), (0, 2), (0, 3), (1, 4), (1, 6), (1, 7), (1, 9)]  

[(k, [tp[1] for tp in tpls if tp[0] == k])
 for k in set([*zip(*tpls)][0])]

Out[11]: [(0, [1, 2, 3]), (1, [4, 6, 7, 9])]

[*zip(*tpls)] 是一个“转置”子迭代的习语

给[(0, 0, 0, 1, 1, 1, 1), (1, 2, 3, 4, 6, 7, 9)]

所以set([*zip(*tpls)][0]) 是set((0, 0, 0, 1, 1, 1, 1))

它给出了tpls中元组第一个位置的唯一项：0, 1

外部for k in ... 迭代，将k 提供给结果元组内的列表comp[tp[1] for tp in tpls if tp[0] == k]