Question

对于字典“a”，键“x、y 和 z”包含整数值。如果我想合并字典中的两个键（考虑到键的大小相同且值为整数类型），生成联合列表的最有效方法是什么？

x+y 和 y+z ？ .

说明：

假设您必须合并两个键并将它们合并到一个新列表或新字典中，而不改变原始字典。

示例：

a = {"x" : {1,2,3,....,50}
     "y" : {1,2,3,.....50}
     "z" : {1,2,3,.....50}
    }

想要的列表：

x+y = [2,4,6,8.....,100]
y+z = [2,4,6,......,100]

Answer 1

您似乎在尝试模仿 join 类型的操作。这不是 Python dicts 的本机，因此如果您真的想要这种类型的功能，我建议您查看 pandas 库。

如果您只想合并没有更高级功能的字典键，此功能应该会有所帮助：

from itertools import chain
from collections import Counter
from typing import Dict, List, Set, Tuple

def merge_keys(data: Dict[str, Set[int]], *merge_list: List[Tuple[str, str]]):
    merged_data = dict()
    merged_counts = Counter(list(chain(*map(lambda k: list(data.get(k, {})) if k in merge_list else [], data))))
    merged_data['+'.join(merge_list)] = [k*v for k,v in merged_counts.items()]
    return merged_data

您可以使用 merge_keys(a, "x", "y", "z", ...) 运行它，其中 a 是您的 dict 的名称 - 您可以根据需要放置任意数量的键（“x”、“y”、“z”、. ..)，因为这个函数接受可变数量的参数。

如果你想在同一个字典中进行两个单独的合并，你需要做的就是：

b = merge_keys(a, "x", "y") | merge_keys(a, "y", "z")

请注意，键的顺序会改变最终合并的键（“y+z”与“z+y”），但不会改变它们合并集合的值。

P.S：这实际上有点棘手，因为原始 dict 设置了值，而不是未排序的列表，因此您不能仅按元素添加它们。这就是我在这里使用 Counter 的原因，以防您想知道。

Answer 2

一种非常有效的方法是将字典转换为 Pandas 数据帧，并允许它使用矢量化方法为您完成工作：

import pandas as pd

a = {"x" : range(1,51), "y" : range(1,51), "z" : range(1,51)}

df = pd.DataFrame(a)
x_plus_y = (df['x'] + df['y']).to_list()
y_plus_z = (df['y'] + df['z']).to_list()

print(x_plus_y)
#[2, 4, 6, 8, 10, 12, 14, 16, 18, 20, 22, 24, 26, 28, 30, 32, 34, 36, 38, 40, 42, 44, 46, 48, 50, 52, 54, 56, 58, 60, 62, 64, 66, 68, 70, 72, 74, 76, 78, 80, 82, 84, 86, 88, 90, 92, 94, 96, 98, 100]

在python中合并单个字典的两个键

2 个答案: