如何通过循环遍历 str 列表来创建 {str, [list of str]} 的字典？

Question

在课程中需要餐厅推荐项目的帮助。

我有一个字符串/值列表，我想通过循环创建一个 {str, list of str} 的字典。这是字符串/值的列表：

contents_list_2 = ['Georgie Porgie', 87, '$$$', 'Canadian,Pub Food', 'Queen St. Cafe', 82, '$', 'Malaysian,Thai', 'Dumplings R Us', 71, '$', 'Chinese', 'Mexican Grill', 85, '$$', 'Mexican', 'Deep Fried Everything', 52, '$', 'Pub Food']

我正在尝试创建一个使用 $ 作为键并使用餐厅名称作为值的字典。例如：

{'$': ['Queen St. Cafe', 'Dumplings R Us', 'Deep Fried Everything'],
 '$$': ['Mexican Grill'],
 '$$$': ['Georgie Porgie'],
 '$$$$': []}

我尝试过使用 zip() 函数，但它没有捕获所有餐厅值：

price_to_name = dict(zip(contents_list_2[2::4], contents_list_2[0::4]))
print(price_to_name)
>>> {'$$$': 'Georgie Porgie', '$': 'Deep Fried Everything', '$$': 'Mexican Grill'}

我认为挑战来自为每个键添加一个值列表...我尝试了各种方法，但成功有限。任何帮助将不胜感激！

Answer 1

由于您的数据以 4 为一组，因此最简单的解决方案可能就是循环它：

from collections import defaultdict

contents_list_2 = ['Georgie Porgie', 87, '$$$', 'Canadian,Pub Food', 'Queen St. Cafe', 82, '$', 'Malaysian,Thai', 'Dumplings R Us', 71, '$', 'Chinese', 'Mexican Grill', 85, '$$', 'Mexican', 'Deep Fried Everything', 52, '$', 'Pub Food']
price_to_name = defaultdict(list)
for i in range(0, len(contents_list_2), 4):
    price_to_name[contents_list_2[i + 2]].append(contents_list_2[i])

price_to_name = dict(price_to_name) # get rid of default value
print(price_to_name)

Answer 2

假设字符串以四个为一组，其中第一个是名称，第三个是 $ 评级，您可以这样做

content = ['Georgie Porgie', 87, '$$$', 'Canadian,Pub Food', 'Queen St. Cafe', 82, '$', 'Malaysian,Thai', 'Dumplings R Us', 71, '$', 'Chinese', 'Mexican Grill', 85, '$$', 'Mexican', 'Deep Fried Everything', 52, '$', 'Pub Food']

ratings = { rating:[N for N,R in zip(content[::4],content[2::4]) if R==rating]
                   for rating in ("$","$$","$$$","$$$$") }

print(ratings)

{'$': ['Queen St. Cafe', 'Dumplings R Us', 'Deep Fried Everything'],
 '$$': ['Mexican Grill'],
 '$$$': ['Georgie Porgie'],
 '$$$$': []}

Answer 3

A collections.defaultdict(list) 是这里的解决方案。您只需要显式循环您的 zip-ed 数据和 append 而不是分配：

from collections import defaultdict  # At top of file

price_to_name = defaultdict(list)
for price, name in zip(contents_list_2[2::4], contents_list_2[::4]):
    price_to_name[price].append(name)

这使工作O(n)（每个切片O(n)，另一个O(n)循环遍历它们，平均情况/摊销O(1)分别用于查找和append，总体上仍然是O(n) )。

如果names 可能重复，并且名称的顺序无关紧要（和/或成员资格测试效率很重要），请将list 更改为set 并将append 更改为add。

从技术上讲，您可以使用常规的 dict 来执行此操作，然后改为使用 price_to_name.setdefault(price, []).append(name)，但这会不必要地一遍又一遍地创建一个空的 list。成员资格测试或try/except KeyError 也可以，但collections.defaultdict 只是最简单、最快和最明显的解决方案。