擅长:python、mysql、java
<p>使用defaultdict是一种非常快速的暴力破解方法。
如果“person”或“friend”是必需的键(输入元组中的第一个人),那么这将在输出字典中添加people及其朋友。在</p>
<pre><code>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"}}
</code></pre>