找到數字子集的分組

Question

可以說我有這些數字集

a = {1, 2, 3} 
b = {2, 3, 4} 
c = {1, 5} 

我想找到集所有不同的數字分組。其結果將是

{1}, {2, 3}, {4}, {5} 

我幼稚的做法，這是不行的，是這樣的：

data = [{1, 2, 3}, {2, 3, 4}, {1, 5}] 
for i in range(1, 5): 
    s = set.intersection(*[x for x in data if i in x]) 
    print(s) 

它返回

set([1]) 
set([2, 3]) 
set([2, 3]) 
set([2, 3, 4]) 

這可以很容易地去重複，但沒有給出預期的結果。

我怎樣才能得到存在於子集中的數字的分組？

Answer 1

你的代碼有兩個問題：

• 你停在5，但range不包括停止讓你不檢查5.
• 如果值僅在一組，你需要創建一個只包含該值的集合。至少你的預期結果看起來好像這是所需的行爲。

所以通過解決這些問題的代碼是這樣的：

data = [{1, 2, 3}, {2, 3, 4}, {1, 5}] 
for i in range(1, 6): 
    useful_sets = [x for x in data if i in x] 
    if len(useful_sets) <= 1: 
     print(set([i])) 
    else: 
     s = set.intersection(*useful_sets) 
     print(s) 

# prints: 
# {1} 
# {2, 3} 
# {2, 3} 
# {4} 
# {5} 

得到一個完整的（而不是複製）的結果，你可以將它們存儲在一組frozensets：

data = [{1, 2, 3}, {2, 3, 4}, {1, 5}] 
res = set() 
for i in range(1, 6): 
    useful_sets = [x for x in data if i in x] 
    if len(useful_sets) <= 1: 
     res.add(frozenset([i])) 
    else: 
     s = set.intersection(*useful_sets) 
     res.add(frozenset(s)) 

print(res) 
# {frozenset({5}), frozenset({4}), frozenset({2, 3}), frozenset({1})} 

哪個（除了排序）應該正是你想要的。