[已解决]求教一个关于Python列表的问题

月汐影

想写一个函数，对一个由列表组成的列表实现合并有重复元素的列表并去掉重复元素。
比如说对[[1,7],[2,5],[5,6]],输出[[1,7],[2,5,6]]...
新手求助，试了半天循环没写出来

最佳答案

月排行榜 / 总排行榜

SixPy

2017-5-10 12:48:03

1. >>> ls = [[1,5],[2,5],[5,6],[1,2]]
   
2. >>> def fn(ls:list):
   
3.         l=[set(ls[0])]
   
4.         for x in ls:
   
5.                 x=set(x)
   
6.                 for s in l:
   
7.                         if s&x: s|=x;break
   
8.                 else:
   
9.                         l.append(x)
   
10.         return [list(x)for x in l]
    
11. 
    
12. >>> fn(ls)
    
13. [[1, 2, 5, 6]]
    
14. >>> fn([[1,7],[2,5],[5,6]])
    
15. [[1, 7], [2, 5, 6]]
    
16. >>>

复制代码

跳转到最佳答案楼层

冬雪雪冬

尝试写了一个，可能没有把所有的可能性都考虑到。

1. import itertools
   
2. list1 = [[1,7],[2,5],[5,6]]
   
3. for i, j in itertools.combinations(list1, 2):
   
4.     if len(set(i) & set(j)) != 0:
   
5.         list1.remove(i)
   
6.         list1.remove(j)
   
7.         list1.append(list(set(i) | set(j)))
   
8. print(list1)

复制代码

月汐影

冬雪雪冬 发表于 2017-5-9 22:24
尝试写了一个，可能没有把所有的可能性都考虑到。

1. list1 = [[1,5],[2,5],[5,6],[1,2]]
   
2.         
   
3. 
   
4. for each1 in list1:
   
5.     for each2 in list1:
   
6.          if each1 != each2:
   
7.              if len(set(each1) & set(each2)) != 0:
   
8.                  list1.remove(each1)
   
9.                  list1.remove(each2)
   
10.                  list1.append(list(set(each1) | set(each2)))
    
11.                  break
    
12. 
    
13. print(list1)
    

复制代码

谢谢版主回答，但是我更换一下列表之后运行就会报错，然后我自己尝试写了下这个，运行结果是[[5, 6], [1, 2, 5]]，这种情况是不是要写迭代？

冬雪雪冬

月汐影 发表于 2017-5-10 05:47
谢谢版主回答，但是我更换一下列表之后运行就会报错，然后我自己尝试写了下这个，运行结果是[[5, 6], ...

你把报错的列表发上来，我再看看。

wei_Y

条件不足吧，如果是 [[1, 2], [3, 4], [2, 3]] 处理后是 [[1, 2, 3, 4]]还是其他的？

SixPy

1. >>> ls = [[1,5],[2,5],[5,6],[1,2]]
   
2. >>> def fn(ls:list):
   
3.         l=[set(ls[0])]
   
4.         for x in ls:
   
5.                 x=set(x)
   
6.                 for s in l:
   
7.                         if s&x: s|=x;break
   
8.                 else:
   
9.                         l.append(x)
   
10.         return [list(x)for x in l]
    
11. 
    
12. >>> fn(ls)
    
13. [[1, 2, 5, 6]]
    
14. >>> fn([[1,7],[2,5],[5,6]])
    
15. [[1, 7], [2, 5, 6]]
    
16. >>>

复制代码

月汐影

wei_Y 发表于 2017-5-10 12:19
条件不足吧，如果是 [[1, 2], [3, 4], [2, 3]] 处理后是 [[1, 2, 3, 4]]还是其他的？

对，就是[[1,2,3,4]]

月汐影

冬雪雪冬 发表于 2017-5-10 08:25
你把报错的列表发上来，我再看看。

没有报错，但是结果不对。。