我有这本包含数据的字典:
{1000: 'brown', 4000: 'yellow', 2000: 'yellow', 3000: 'blue'}
然而,我的字典有更多的数据,但这就是结构。键是唯一的,但值不是,因为它可能适用于多个键,我之前使用了一个循环来确保其中没有重复的键。
我想要的是将字典分类为列表/字典/图列,无论什么工作。所以一份全棕色的清单,一份全黄色的清单,一份全蓝色的清单(只有三种颜色,所以我需要三份清单)。有点像这样:
brown_list = [1000, 5000, 7000]
yellow_list = [4000, 2000, 8000]
blue_list = [3000, 9000, 10000]
我试图这样做(color_name 是我的原始字典):
for color in color_name:
if color_name == 'brown'
brown_list.append color_name
不幸的是,这不起作用。我试图阅读字典并从它们中附加,但我并没有真正理解它。关于如何根据 dict 的值将字典拆分为不同列表的任何提示?
答案 0 :(得分:2)
您应该使用更简单的访问方法来反映您的结构:另一个以颜色为键的字典:
>>> value_dict = {1000: 'brown', 4000: 'yellow', 2000: 'yellow', 3000: 'blue'}
>>> set(value_dict.values())
{'brown', 'blue', 'yellow'}
>>> color_dict = {}
>>> for color in set(value_dict.values()):
color_dict[color] = [v for v, c in value_dict.items() if c == color]
>>> color_dict
{'brown': [1000], 'blue': [3000], 'yellow': [4000, 2000]}
如果您想获得 uber-Pythonic,您可以将转换折叠为一行,即字典推导中的列表推导:
color_dict{color: [v for v, c in value_dict.items() if c == color]
for color in set(value_dict.values())
}
答案 1 :(得分:1)
append
是一个方法;方法调用需要括号for number, color in your_dictionary.items():
if color == 'brown':
brown_list.append(number)
elif color == 'yellow':
...
答案 2 :(得分:1)
您可以创建一个输出字典,其键是颜色,其关联值是与每种颜色关联的数据值列表。
例如:
dict1 = {1000: 'brown', 4000: 'yellow', 2000: 'yellow', 3000: 'blue'}
dict2 = {}
for k,v in dict1.items():
dict2[v] = [*dict2.get(v, []), k]
print(dict2)
结果是:{'brown': [1000], 'yellow': [4000, 2000], 'blue': [3000]}
这可以更好地缩放其他颜色。您可以使用 blue_list
代替 dict2['blue']
。
答案 3 :(得分:1)
将数字存储在字典中,并使用 collections.defaultdict
创建它:
from collections import defaultdict
color_name = {1000: 'brown', 4000: 'yellow', 2000: 'yellow', 3000: 'blue', 5000: 'brown', }
color_to_nums = defaultdict(list)
for num, color in color_name.items():
color_to_nums[color] += [num]
# Optional: convert back into regular dictionary
color_to_nums = dict(color_to_nums)
print(color_to_nums)
# {'brown': [1000, 5000], 'yellow': [4000, 2000], 'blue': [3000]}