从字典追加到列表?

时间:2021-04-05 19:27:58

标签: python dictionary

我有这本包含数据的字典:

{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 的值将字典拆分为不同列表的任何提示?

4 个答案:

答案 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 是一个方法;方法调用需要括号
  • 在 if 语句后加一个冒号
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]}