算法 - Python二维数组怎么求并集?

Question

a=[[1,2,'g'],[2,4,'f']]
b=[[2,4,'f'],[1,2,'f']]

a和b的并集是

c=[[1,2,'g'],[2,4,'f'],[1,2,'f']]

无法使用set().union,怎么破?只能自己实现遍历么?

Answer 1

你可以用以下程式碼來實現：

a=[[1,2,'g'],[2,4,'f']]
b=[[2,4,'f'],[1,2,'f']]
tmp = a + b
c = [ (tmp[i]) for i in range(0, len(tmp)) if tmp[i] not in tmp[:i] ]

檢查結果：

print c
[[1, 2, 'g'], [2, 4, 'f'], [1, 2, 'f']]

Answer 2

可以這樣：

a=[[1,2,'g'],[2,4,'f']]
b=[[2,4,'f'],[1,2,'f']]
c=[list(i) for i in set(tuple(j) for j in a+b)]

檢查結果：

print c
[[2, 4, 'f'], [1, 2, 'f'], [1, 2, 'g']]

Answer 3

直接使用集合操作肯定是不行的，因為列表裡面的元素還是列表，是一個可變對象，而集合操作的對象必須是不可變對象，因此如果想去重的話， 簡單點的方法就是樓上的回复，轉換為不可變物件tuple,從而再進行set去重。 如下，其實還是遍歷。

map(list,set(map(tuple, a + b)))