元组对列表到字典中

Question

我在使用 Python 中的字典时遇到问题。我意识到这个任务很简单，但我无法弄清楚。我的任务是编写一个函数，它将元组列表转换为字典，其中字典的键代表人（来自元组），值代表他们的朋友。每经验：

pairs_of_people = {("Adam", "Brian"), ("Adam", "Gabe"), ("Adam", "Hagan"),
    ("Brian", "Calvin"), ("Brian", "Hagan"),
    ("Calvin", "Dan")}`

必须返回：

{"Adam": {"Brian", "Gabe", "Hagan"}, "Brian": {"Adam", "Hagan", "Calvin"}, "Calvin": {"Brian", "Dan"}}

有什么帮助吗？谢谢你。如果有人对某处的字典有很好的介绍，将不胜感激。我就是想不通。

Answer 1

暴力破解此解决方案的一种非常快速的方法是使用默认字典。 如果“人”或“朋友”是必需的键（输入元组中的第一个人），这将把人和他们的朋友添加到我们的输出字典中。

from collections import defaultdict

pairs_of_people = {("Adam", "Brian"), ("Adam", "Gabe"), ("Adam", "Hagan"),
               ("Brian", "Calvin"), ("Brian", "Hagan"),
               ("Calvin", "Dan")}

required_keys = {t[0] for t in pairs_of_people}

out = defaultdict(set)

for person, friend in pairs_of_people:
    if person in required_keys:
        out[person].add(friend)
    if friend in required_keys:
        out[friend].add(person)

print(out)
>> {"Adam": {"Brian", "Gabe", "Hagan"}, "Brian": {"Adam", "Hagan", "Calvin"},
    "Calvin": {"Brian", "Dan"}}

Answer 2

通过使用defaultdict 和set 作为默认工厂，构造这个包含sets 的字典（显然）要容易得多：

from collections import defaultdict

d = defaultdict(set) 

另外，使用names 的辅助set 包含每个元组的名字。这样您就可以检查成员资格，如果有两对是 names 集中的名称，它们将被添加到彼此的结果集中：

names = {i[0] for i in pairs_of_people}

for i, j in pairs_of_people:
    d[i].add(j)
    if j in names:
        d[j].add(i)

d = dict(d)   # if necessarily a dict

defaultdict 只是一种特殊的字典，可以让这些任务变得不那么乏味。现在，d 包含：

print(d)
{'Brian': {'Hagan', 'Adam', 'Calvin'}, 'Adam': {'Hagan', 'Brian', 'Gabe'}, 'Calvin': {'Brian', 'Dan'}}

由于您的元组包含在一个集合中，因此除非您将它们移动到一个列表然后对它们进行排序，否则您无法进行预先确定的排序。您还可以更改defaultdict 以使用d = defaultdict(list) 的默认列表工厂，并使用d[i].append(j) 和d[j].append(i) 而不是.add。

关于字典的介绍，可以看Python Tutorial的字典部分。