按键后缀拆分字典

Question

我有这样的字典

d = {"key_a":1, "anotherkey_a":2, "key_b":3, "anotherkey_b":4}

所以值和键名在这里并不重要。 关键（没有双关语意）是相关的键在我上面的示例中共享相同的后缀，即_a和_b 。

这些后缀是事先不知道的（例如，它们并不总是_a和_b ，并且有未知数量的不同后缀。

我想做的是将相关的键提取到他们自己的字典中，并将所有生成的字典放在一个列表中。 上面的 output 是

output = [{"key_a":1, "anotherkey_a":2},{"key_b":3, "anotherkey_b":4}]

我目前的做法是先获取所有的后缀，然后一次生成一个sub-dicts和append到新列表

output = list()
# Generate a set of suffixes
suffixes = set([k.split("_")[-1] for k in d.keys()])

# Create the subdict and append to output
for suffix in suffixes:
    output.append({k:v for k,v in d.items() if k.endswith(suffix)})

这有效（并且不是非常慢或任何东西）但我只是想知道是否有更优雅的方法来使用列表或字典理解来做到这一点？ 只是出于兴趣...

Answer 1

使您的输出成为 defaultdict 而不是列表，后缀为键：

from collections import defaultdict
output = defaultdict(lambda: {})
for k, v in d.items():
    prefix, suffix = k.rsplit('_', 1)
    output[suffix][k] = v

这将在一次传递中拆分您的 dict 并导致如下结果：

输出 = {"a" : {"key_a":1, "anotherkey_a":2}, "b": {"key_b":3, "anotherkey_b":4}}

如果您坚持将其转换为列表，则可以简单地使用：

output = list(output.values())

Answer 2

你可以压缩线条

output = list()

for suffix in suffixes:
    output.append({k:v for k,v in d.items() if k.endswith(suffix)})

到列表理解，像这样

[{k:v for k,v in d.items() if k.endswith(suffix)} for suffix in suffixes]

是否更优雅，恐怕在旁人眼里。

不过，@Błotosmętek 建议的方法可能会更快，因为它有一个大字典，因为它会减少循环。

Answer 3

def sub_dictionary_by_suffix(dictionary, suffix):
    sub_dictionary = {k: v for k, v in dictionary.items() if k.endswith(suffix)}

    return sub_dictionary

我希望它有帮助