在python中合并字典值列表
Posted
技术标签:
【中文标题】在python中合并字典值列表【英文标题】:Merging dictionary value lists in python 【发布时间】:2015-01-10 16:56:59 【问题描述】:我正在尝试合并三个字典,它们都具有相同的键、值列表或单个值。
one='a': [1, 2], 'c': [5, 6], 'b': [3, 4]
two='a': [2.4, 3.4], 'c': [5.6, 7.6], 'b': [3.5, 4.5]
three='a': 1.2, 'c': 3.4, 'b': 2.3
我需要将值中的所有项目添加到一个列表中。
result='a': [1, 2, 2.4, 3.4, 1.2], 'c': [5, 6, 5.6, 7.6, 2.3], 'b': [3, 4, 3.5, 4.5, 3.4]
我尝试了几件事,但大多数都将值放入嵌套列表中。 例如
out=dict((k, [one[k], two.get(k), three.get(k)]) for k in one)
'a': [[1, 2], [2.4, 3.4], 1.2], 'c': [[5, 6], [5.6, 7.6], 3.4], 'b': [[3, 4], [3.5, 4.5], 2.3]
我尝试通过遍历值来更新它:
out.update((k, [x for x in v]) for k,v in out.iteritems())
但结果完全一样。 我试图简单地添加列表,但是因为第三个字典只有一个浮点数,所以我做不到。
check=dict((k, [one[k]+two[k]+three[k]]) for k in one)
所以我尝试先将列表添加到值 1 和 2,然后附加值 3。添加列表效果很好,但是当我尝试从第三个字典中附加浮点数时,突然整个值变为“无”
check=dict((k, [one[k]+two[k]]) for k in one)
'a': [[1, 2, 2.4, 3.4]], 'c': [[5, 6, 5.6, 7.6]], 'b': [[3, 4, 3.5, 4.5]]
new=dict((k, v.append(three[k])) for k,v in check.items())
'a': None, 'c': None, 'b': None
【问题讨论】:
【参考方案1】:作为一个单行,具有字典理解:
new = key: value + two[key] + [three[key]] for key, value in one.iteritems()
这会创建新列表,将来自one
的列表与来自two
的相应列表连接起来,将three
中的单个值放入一个临时列表中,以便于连接。
或者使用for
循环更新one
就地:
for key, value in one.iteritems():
value.extend(two[key])
value.append(three[key])
这使用list.extend()
用来自two
的列表就地更新原始列表,并使用list.append()
来添加来自three
的单个值。
你哪里出错了:
您的第一次尝试创建一个新列表,其中包含来自 one
、two
和 three
的值嵌套在其中而不是连接现有列表。您尝试清理它只是复制了那些嵌套列表。
您的第二次尝试没有成功,因为 three
中的值不是一个列表,因此无法连接。我只为那个值创建了一个新列表。
您的最后一次尝试不应该在生成器表达式中使用list.append()
,因为您存储了该方法的返回值,它始终是None
(它的更改存储在直接v
,列表不需要再次返回)。
第一种方法的演示:
>>> one='a': [1, 2], 'c': [5, 6], 'b': [3, 4]
>>> two='a': [2.4, 3.4], 'c': [5.6, 7.6], 'b': [3.5, 4.5]
>>> three='a': 1.2, 'c': 3.4, 'b': 2.3
>>> key: value + two[key] + [three[key]] for key, value in one.iteritems()
'a': [1, 2, 2.4, 3.4, 1.2], 'c': [5, 6, 5.6, 7.6, 3.4], 'b': [3, 4, 3.5, 4.5, 2.3]
【讨论】:
对于python3
使用items()
而不是iteritems()
【参考方案2】:
任意字典编号和键
@MartijnPieters' solution 涵盖了您尝试的问题。
对于通用解决方案,考虑使用itertools.chain
链接多个字典。您也可以使用defaultdict
来处理在每个字典中找不到相同键的更一般情况。
from collections import defaultdict
from itertools import chain
from operator import methodcaller
# dictionaries with non-equal keys, values all lists for simplicity
one = 'a': [1, 2], 'c': [5, 6], 'b': [3, 4], 'e': [6.2]
two = 'a': [2.4, 3.4], 'c': [5.6, 7.6], 'b': [3.5, 4.5], 'f': [1.3]
three = 'a': [1.2], 'c': [3.4], 'b': [2.3], 'e': [3.1]
# initialise defaultdict of lists
dd = defaultdict(list)
# iterate dictionary items
dict_items = map(methodcaller('items'), (one, two, three))
for k, v in chain.from_iterable(dict_items):
dd[k].extend(v)
print(dd)
# defaultdict(list,
# 'a': [1, 2, 2.4, 3.4, 1.2],
# 'b': [3, 4, 3.5, 4.5, 2.3],
# 'c': [5, 6, 5.6, 7.6, 3.4],
# 'e': [6.2, 3.1],
# 'f': [1.3])
注意defaultdict
是dict
的子类,因此通常不需要将结果转换为常规dict
。
【讨论】:
【参考方案3】:一个强大的解决方案。 =)
def FullMergeDict(D1, D2):
for key, value in D1.items():
if key in D2:
if type(value) is dict:
FullMergeDict(D1[key], D2[key])
else:
if type(value) in (int, float, str):
D1[key] = [value]
if type(D2[key]) is list:
D1[key].extend(D2[key])
else:
D1[key].append(D2[key])
for key, value in D2.items():
if key not in D1:
D1[key] = value
if __name__ == '__main__':
X =
'a': 'aaa',
'c': [1,3,5,7],
'd': 100,
'e': 'k': 1, 'p': 'aa','t': [-1,-2],
'f': 'j':1
Y =
'b': 'bbb',
'd': 200,
'e': 'k': 2, 'p': 'bb','o': [-4],
'c': [2,4,6],
'g': 'v':2
FullMergeDict(X, Y)
exit(0)
结果:
X =
'a': 'aaa',
'b': 'bbb',
'c': [1, 3, 5, 7, 2, 4, 6],
'd': [100, 200],
'e': 'k': [1, 2], 'o': [-4], 'p': ['aa', 'bb'], 't': [-1, -2],
'f': 'j': 1,
'g': 'v': 2
【讨论】:
优秀的解决方案!我只是用它来合并两个列表字典,其中列表值是元组对的元组。哦,两个字典中的键也是元组。所有元组(键和值)都是坐标对。像冠军一样工作!【参考方案4】:如果您在字典中有不同的键和不同类型的值,您可以使用以下方法:
from collections import defaultdict, Iterable
dct1 = 'a': [1, 2]
dct2 = 'a': [3], 'b': [5, 6]
dct3 = 'a': 4, 'c': 7
result = defaultdict(list)
for dct in [dct1, dct2, dct3]:
for k, v in dct.items():
if isinstance(v, Iterable):
result[k].extend(v)
else:
result[k].append(v)
print(result)
# defaultdict(<class 'list'>, 'a': [1, 2, 3, 4], 'b': [5, 6], 'c': [7])
【讨论】:
【参考方案5】:单行解决方案(也处理仅存在于一个字典中的键):
key:one.get(key,[])+two.get(key,[]) for key in set(list(one.keys())+list(two.keys()))
示例 1:
one = 'a': [1, 2], 'c': [5, 6], 'b': [3, 4]
two = 'a': [2.4, 3.4], 'c': [5.6, 7.6], 'd': [3.5, 4.5]
key:one.get(key,[])+two.get(key,[]) for key in set(list(one.keys())+list(two.keys()))
输出:
'a': [1, 2, 2.4, 3.4], 'b': [3, 4], 'c': [5, 6, 5.6, 7.6], 'd': [3.5, 4.5]
示例 2:
x=1:['a','b','c']
y=1:['d','e','f'],2:['g']
key:x.get(key,[])+y.get(key,[]) for key in set(list(x.keys())+list(y.keys()))
输出:
1: ['a', 'b', 'c', 'd', 'e', 'f'], 2: ['g']
【讨论】:
【参考方案6】:是否查看此帮助:
>>> dic=
>>> k=[]
>>> for i in 'abc':
k=one[i]+two[i]
k.append(three[i])
dic[i]=k
>>> dic
'c': [5, 6, 5.6, 7.6, 3.4], 'a': [1, 2, 2.4, 3.4, 1.2], 'b': [3, 4, 3.5, 4.5, 2.3]
【讨论】:
以上是关于在python中合并字典值列表的主要内容,如果未能解决你的问题,请参考以下文章
Python列表元素为字典时,如何根据其中某个相同的键值进行元素合并