从字典中的列表中有效地提取一组唯一值

问题描述

我有一个看起来像这样的数据结构：

{'A': [2,3,5,6],'B': [1,2,4,7],'C': [1,'D': [1,'E': [3,4]}

使用Python，我需要提取以下内容：

{1,6,7}

因为我需要对下游的数学方程计算不同的值。

这是我当前的实现，可以正常工作（完整的代码示例）：

from itertools import chain

# Create som mock data for testing
dictionary_with_lists = {'A': [2,4]}

print(dictionary_with_lists)

# Output: 'A': [2,4]}

# Flatten dictionary to list of lists,discarding the keys
list_of_lists = [dictionary_with_lists[i] for i in dictionary_with_lists]
print(f'list_of_lists: {list_of_lists}')

# Output: list_of_lists: [[2,[1,[3,4]]

# Use itertools to flatten the list
flat_list = list(chain.from_iterable(list_of_lists))
print(f'flat_list: {flat_list}')

# Output: flat_list: [2,1,7,4]

# Convert list to set to get only unique values
set_of_unique_items = set(flat_list)
print(f'set_of_unique_items: {set_of_unique_items}')

# Output: set_of_unique_items: {1,7}

虽然可行，但我怀疑可能会有一种更简单，更有效的方法。

在不降低代码可读性的情况下，什么是更有效的实现？

我的现实世界词典包含成千上万个任意长度的列表。

解决方法

一个人的观点：

https

,

尝试一下

from itertools import chain

d = {'A': [2,3,5,6],'B': [1,2,4,7],'C': [1,'D': [1,'E': [3,4]}
print(set(chain.from_iterable(d.values())))

输出：

{1,6,7}

,

s = set()
for key in dictionary_with_lists:
    for val in dictionary_with_lists[key]:
        s.add(val)