[英]Efficiently extracting set of unique values from lists within a dictionary
I have a data structure which looks like this:我有一个看起来像这样的数据结构:
{'A': [2, 3, 5, 6], 'B': [1, 2, 4, 7], 'C': [1, 3, 4, 5, 7], 'D': [1, 4, 5, 6], 'E': [3, 4]}
Using Python, I need to extract this:使用 Python,我需要提取这个:
{1, 2, 3, 4, 5, 6, 7}
Because I need a count of the distinct values for a mathematical equation further downstream.因为我需要计算更下游的数学方程式的不同值。
Here is my current implementation, which works (complete code example):这是我当前的实现,它有效(完整的代码示例):
from itertools import chain
# Create som mock data for testing
dictionary_with_lists = {'A': [2, 3, 5, 6],
'B': [1, 2, 4, 7],
'C': [1, 3, 4, 5, 7],
'D': [1, 4, 5, 6],
'E': [3, 4]}
print(dictionary_with_lists)
# Output: 'A': [2, 3, 5, 6], 'B': [1, 2, 4, 7], 'C': [1, 3, 4, 5, 7], 'D': [1, 4, 5, 6], 'E': [3, 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, 3, 5, 6], [1, 2, 4, 7], [1, 3, 4, 5, 7], [1, 4, 5, 6], [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, 3, 5, 6, 1, 2, 4, 7, 1, 3, 4, 5, 7, 1, 4, 5, 6, 3, 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, 2, 3, 4, 5, 6, 7}
While this works, but I suspect there might be a simpler and more efficient approach.虽然这可行,但我怀疑可能有更简单、更有效的方法。
What would be a more efficient implementation which does not diminish code readability?什么是不降低代码可读性的更有效的实现?
My real-world dictionary contains hundreds of thousands or millions of lists of arbitrary lengths.我的真实世界词典包含数十万或数百万个任意长度的列表。
An outsdider's point of view:一个局外人的观点:
dict = {'A': [2, 3, 5, 6], 'B': [1, 2, 4, 7], 'C': [1, 3, 4, 5, 7], 'D': [1, 4, 5, 6], 'E': [3, 4]}
S = set()
for L in dict.values():
S = S.union(set(L))
Try this试试这个
from itertools import chain
d = {'A': [2, 3, 5, 6], 'B': [1, 2, 4, 7], 'C': [1, 3, 4, 5, 7], 'D': [1, 4, 5, 6], 'E': [3, 4]}
print(set(chain.from_iterable(d.values())))
Output: Output:
{1, 2, 3, 4, 5, 6, 7}
s = set()
for key in dictionary_with_lists:
for val in dictionary_with_lists[key]:
s.add(val)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.