假设在一个列表number_pairs中有一系列两个两个数字组成的子列表,该子列表代表着列表中两个数字存在关联(可以把这个关系写作子列表中第三个元素,比如[1,2,'关系a'] 但因为跟需求没啥关系,故省略),现在需要的是把所有能通过直接/间接关联的子列表放在一个新列表中,并把没有关联的子列表单独列出。
比如:一个 number_pairs可以长成下面这样
number_pairs = [[1, 0], [2, 3], [1, 2], [5, 8], [3, 4], [1, 4], [6, 9]]
在这里我们可以看到 [[1, 0], [2, 3],[1, 2],[3, 4],[1, 4]]这五个子列表中的任意一个是可以通过直接/间接关联找到其他子列表的的(比如[1,0], [2,3] 两个子列表本身没关系,因为都和[1,2] 有共用元素,所以彼此也有了关系),但是[5, 8], [6, 9]这两个子列表则无法与其他任何子列表产生联系。
目前写了一段一边循环一边更新列表的代码如下
info = []
for i in number_pairs:
relation = []
relation.append(i)
number_pairs.remove(i)
for j in relation:
[a, b] = j
for k in number_pairs:
if a in k or b in k:
relation.append(k)
number_pairs.remove(k)
info.append(relation)
# 结果:
# info
# [[[1, 0], [1, 2], [1, 4], [2, 3], [3, 4]], [[6, 9]]], 可以看见 [5,8]被跳过了
逻辑是先循环number_pairs中找到其中的一个单位i并将其从列表中剔除,然后再循环剩下的列表元素j,判断i中元素a,b是否有一个与j有关联,如果有关联就加入relation并剔除number_pairs
实现出来会稳定地跳过每次info.append(relation)完成后的下一项子列表
比如 [5, 8] 就在这里被跳过了,如果在原number_pair的[5,8]前加入一个 [10,0], 那么[10,0]被跳过的同时,最后的[6, 9]也会被跳过
虽然有听说过修改一个正在被遍历的列表是在作死,但是暂时没想到别的方法 求指教