为字典中的一个键追加多个值

Question

我是 python 新手，我有每年的年份和值列表。我想要做的是检查字典中是否已存在该年份，如果存在，则将该值附加到特定键的值列表中。

例如，我有一个年份列表，每年都有一个值：

2010  
2  
2009  
4  
1989  
8  
2009  
7  

我想要做的是以年份作为键，以那些个位数作为值来填充字典。但是，如果我将 2009 年列出两次，我想将第二个值附加到该字典中的值列表中，所以我想要：

2010: 2  
2009: 4, 7  
1989: 8  

现在我有以下内容：

d = dict()  
years = []  

(get 2 column list of years and values)

for line in list:    
    year = line[0]   
    value = line[1]  

for line in list:  
    if year in d.keys():  
        d[value].append(value)  
    else:  
        d[value] = value  
        d[year] = year  

Answer 1

如果我可以重新表述你的问题，你想要的是一个以年份为键的字典，以及每年的一个数组，其中包含与该年份相关的值列表，对吧？我是这样做的：

years_dict = dict()

for line in list:
    if line[0] in years_dict:
        # append the new number to the existing array at this slot
        years_dict[line[0]].append(line[1])
    else:
        # create a new array in this slot
        years_dict[line[0]] = [line[1]]

在years_dict中你最终应该得到一个如下所示的字典：

{
    "2010": [2],
    "2009": [4,7],
    "1989": [8]
}

一般来说，创建“并行数组”是一种糟糕的编程实践，其中项目通过具有相同的索引，而不是包含它们的容器的正确子项。

Answer 2

您最好使用 collections.defaultdict （添加在Python 2.5）。这允许您指定缺失键的默认对象类型（例如 list）。

因此，您不需要先创建一个不存在的键，然后附加到该键的值，而是省去中间人，直接附加到不存在的键以获得所需的结果。

使用数据的简单示例：

>>> from collections import defaultdict
>>> data = [(2010, 2), (2009, 4), (1989, 8), (2009, 7)]
>>> d = defaultdict(list)
>>> d
defaultdict(<type 'list'>, {})
>>> for year, month in data:
...     d[year].append(month)
... 
>>> d
defaultdict(<type 'list'>, {2009: [4, 7], 2010: [2], 1989: [8]})

这样您就不必担心是否看到与年份相关的数字。您只需添加并忘记，因为知道丢失的键将永远是一个列表。如果一个键已经存在，那么它只会被附加到。

Answer 3

您可以使用setdefault。

for line in list:  
    d.setdefault(year, []).append(value)

这是有效的，因为 setdefault 返回列表并将其设置在字典中，并且因为列表是可变的，所以附加到 setdefault 返回的版本与将其附加到字典本身内的版本相同。如果这有任何意义的话。

Answer 4

d = {} 

# import list of year,value pairs

for year,value in mylist:
    try:
        d[year].append(value)
    except KeyError:
        d[year] = [value]

Python 之道——获得宽恕比请求许可更容易！

Answer 5

这是使用 not in 运算符执行此操作的另一种方法：

# define an empty dict
years_dict = dict()

for line in list:
    # here define what key is, for example,
    key = line[0]
    # check if key is already present in dict
    if key not in years_dict:
        years_dict[key] = []
    # append some value 
    years_dict[key].append(some.value)

Answer 6

如果将这些值放入元组列表中会更容易。为此，您可以使用列表切片和 zip 函数。

data_in = [2010,2,2009,4,1989,8,2009,7]
data_pairs = zip(data_in[::2],data_in[1::2])

Zip 获取任意数量的列表（在本例中为 data_in 的偶数和奇数条目），并将它们放在一起形成一个元组。

现在我们可以使用setdefault方法。

data_dict = {}
for x in data_pairs:
    data_dict.setdefault(x[0],[]).append(x[1])

setdefault 接受一个键和一个默认值，并返回关联值，或者如果没有当前值，则返回默认值。在这种情况下，我们将获得一个空列表或填充列表，然后将当前值附加到其中。

Answer 7

如果您想要（几乎）一行：

from collections import deque

d = {}
deque((d.setdefault(year, []).append(value) for year, value in source_of_data), maxlen=0)

使用 dict.setdefault ，您可以将“检查密钥是否已存在，如果不存在则创建一个新列表”的想法封装到单个调用中。这允许您编写一个生成器表达式，该表达式由 deque 尽可能高效地使用，因为队列长度设置为零。双端队列将立即被丢弃，结果将在d中。

这只是我为了好玩而做的事情。我不建议使用它。在某个时间和地点可以通过双端队列消耗任意可迭代对象，但这绝对不是这样。