3

对于字典“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]
4

2 回答 2

1

似乎您正在尝试模仿join-type 操作。这不是 python dicts 所固有的，所以如果你真的想要那种类型的功能，我建议你看看这个pandas库。

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

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”），但不会更改它们合并集的值。

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

于 2021-03-12T00:37:49.087 回答
1

一种非常有效的方法是将字典转换为 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]
于 2021-03-12T00:47:35.420 回答