假设我有字典:
dict1 = {'A': 1, 'B': 2, 'C' : 3}
dict2 = {'a': 4, 'b': 5, 'c' : 6}
这个链接建议了几种合并两者的方法,但所有的合并都是简单的连接。我想把它们合并起来,就像庄家洗牌一样,或者像拉链拉上一样。我的意思是,一旦合并dict1
和dict2
,得到的dict3
应该变成
dict3 = {'A': 1, 'a': 4, 'B': 2, 'b': 5, 'C' : 3, 'c' : 6}
因此合并以交替的方式从dict1
和dict2
中获取元素。我的字典实际上非常大,所以手动操作是不可行的。
有一个循环的itertools
配方来选择这种方式的数据
你可以使用:
dict3 = dict(roundrobin(dict1.items(), dict2.items()))
输出:
{'A': 1, 'a': 4, 'B': 2, 'b': 5, 'C': 3, 'c': 6}
配方:
from itertools import cycle, islice
def roundrobin(*iterables):
"roundrobin('ABC', 'D', 'EF') --> A D E B F C"
# Recipe credited to George Sakkis
num_active = len(iterables)
nexts = cycle(iter(it).__next__ for it in iterables)
while num_active:
try:
for next in nexts:
yield next()
except StopIteration:
# Remove the iterator we just exhausted from the cycle.
num_active -= 1
nexts = cycle(islice(nexts, num_active))
也可以使用more-itertools.interleave
from more_itertools import interleave
dict(interleave(dict1.items(), dict2.items()))
缩进字号:
from itertools import chain
dict(
chain.from_iterable(
zip(dict1.items(), dict2.items())
)
)
如果您的字典不能保证大小相等,请调整此解决方案以使用itertools.zip_longest()
而不是基本的zip()
这种方法有很多可能出错的地方。如果两个字典的长度不相同怎么办?如果他们有重复的钥匙怎么办?如果以后你需要3个字典而不是2个呢?
同样,在这个需求背后你有一个真实的用例吗?您真的需要像这样用所有数据构建一个新的字典吗?还是仅仅需要像这样以一种特定的替代方式遍历(键、值)对?
假设您不必担心上述问题,如果您只需要迭代,则可以简单地执行如下操作:
def iter_zip(dict1, dict2):
for i1, i2 in zip(dict1.items(), dict2.items()):
yield i1
yield i2
dict1 = {'A': 1, 'B': 2, 'C' : 3}
dict2 = {'a': 4, 'b': 5, 'c' : 6}
# you can then do
for key, val in iter_zip(dict1, dict2):
do_something(key, val)
如果你真的需要建立一个新的字典,现在就像这样简单:
dict(iter_zip(dict1, dict2))
# {'A': 1, 'a': 4, 'B': 2, 'b': 5, 'C': 3, 'c': 6}