我有一个设备字典,每个设备都有一个可以与之协作的其他设备的列表。可以将这些视为与其兼容的设备。
devices = {0: [], 1: [9, 10, 14], 2: [9, 11, 14], 3: [], 4: [], 5: [9, 14, 15], 6: [], 7: [], 8: [], 9: [1, 2, 5, 11, 14, 15], 10: [1], 11: [2, 9, 14], 12: [], 13: [], 14: [1, 2, 5, 9, 11, 16], 15: [5, 9], 16: [14], 17: [], 18: [], 19: []}我希望能够将这些群集到一个列表列表中,该列表包含所有可能的群集,这些群集可以一起协作。一个集群只能包含所有相互兼容的设备。换句话说,如果集群1,9,2存在,那么1必须同时与9和2协作,并且它们也必须相互协作。在这种情况下,最终结果应该如下所示:
[ [1, 9, 14], [2, 9, 14], [5,9,14], [5,15,9] [9,11,14,2], [10,1], [14,16] ]我可能手动计算出了一个错误,但我相信这是所有可能的项目集群,同时满足了它们的兼容性要求。
但是,我在将其转换为代码时遇到了一些困难。任何帮助都将不胜感激。
发布于 2020-11-11 08:11:15
可能有一种更干净的方法来获得结果,但这段代码可以工作。
请注意,您的结果包括2,9,14,这是9,11,14,2的子集。该子集在此代码中被删除。
items = {0: [], 1: [9, 10, 14], 2: [9, 11, 14], 3: [], 4: [], 5: [9, 14, 15], 6: [], 7: [], 8: [],
9: [1, 2, 5, 11, 14, 15], 10: [1], 11: [2, 9, 14], 12: [], 13: [], 14: [1, 2, 5, 9, 11, 16],
15: [5, 9], 16: [14], 17: [], 18: [], 19: []}
grps = []
# create 1-1 groups
for g in items:
for c in items[g]:
if g in items[c]:
grps.append([g,c])
# add other elements to each group
chg = True
while chg: # until no more elements added
chg = False
for g in items: # each single element
for g2 in grps: # check each existing group
if g in g2: continue # if not already in group
ok = True
for c in g2: # check each group member
if not (g in items[c] and c in items[g]): # can we collaborate?
ok = False # no
if ok:
g2.append(g) # add element to group
chg = True
# check for subsets
for i,x in enumerate(grps):
for j,y in enumerate(grps):
if i==j: continue # same group
if set(x) & set(y) == set(x): # if subset
x.clear() # remove elements
grps = [g for g in grps if len(g)] # remove empty groups
print(grps)输出
[[10, 1], [14, 5, 9], [14, 9, 1], [14, 11, 2, 9], [15, 9, 5], [16, 14]]发布于 2020-11-11 09:12:05
可以这样做的一种方法是:
from itertools import combinations
colaborate2 = lambda args, dic : all(x in dic[y] and y in dic[x] for x, y in combinations(args,2))
def group (x, dic, n=None):
if n is None:
n = len(x)
for i in combinations(x, n):
if colaborate2(i, dic):
return tuple(sorted(i))
if n > 1:
return group(x, dic, n - 1)
def groupings(dic):
m = set([group(val + [key], dic) for key, val in dic.items()])
return [i for i in m if len(i)>1]
groupings(items)
[(5, 9, 15), (14, 16), (5, 9, 14), (1, 9, 14), (1, 10), (2, 9, 11, 14)]发布于 2020-11-11 22:52:12
虽然花了我几个小时,但我对整洁干净的解决方案很满意。在编码的同时理解需求花费了我90%的时间。学到的教训--先用铅笔和纸。
代码:
devices = {0: [], 1: [9, 10, 14], 2: [9, 11, 14], 3: [], 4: [], 5: [9, 14, 15], 6: [], 7: [], 8: [], 9: [1, 2, 5, 11, 14, 15], 10: [1], 11: [2, 9, 14], 12: [], 13: [], 14: [1, 2, 5, 9, 11, 16], 15: [5, 9], 16: [14], 17: [], 18: [], 19: []}
# manually calculated expected result [ [1, 9, 14], [2, 9, 14], [5,9,14], [5,15,9] [9,11,14,2], [10,1], [14,16] ]
# true result [[1, 9, 14], [2, 9, 11, 14], [5, 9, 14], [10, 1], [15, 5, 9], [16, 14]]
def is_compatible(k,v):
if v in devices[k]:
return True
return False
clusters = []
for k,v_list in devices.items():
if v_list:
cluster = []
cluster.append(k)
cluster.append(v_list[0])
for v_ele in v_list[1:]:
v_ele_further_check = True
for cluster_member in cluster[1:]: # check v_ele compatible with existing cluster members
if not is_compatible(v_ele, cluster_member):
v_ele_further_check = False
break
if v_ele_further_check:
cluster.append(v_ele)
clusters.append(cluster)
print(clusters)
unique_clusters = []
for cluster in clusters: # eliminate_duplicates
if not sorted(cluster) in unique_clusters:
unique_clusters.append(cluster)
print(unique_clusters)输出:
[[1, 9, 14], [2, 9, 11, 14], [5, 9, 14], [9, 1, 14], [10, 1], [11, 2, 9, 14], [14, 1, 9], [15, 5, 9], [16, 14]]
[[1, 9, 14], [2, 9, 11, 14], [5, 9, 14], [10, 1], [15, 5, 9], [16, 14]]评论:
你可以看到我可以很容易的去掉is_compatible函数,但是它增加了可读性。如果这很重要,集群中的集群的顺序将得到正确的维护。
https://stackoverflow.com/questions/64778200
复制相似问题