python两个列表进行合并

Posted 2023-04-04

例如有a=[[A,A1],[B,A2],[C,A3],[D,A4]]有另一个列表B=[[A,B1],[B,B2],[C,B3],[D,B4]]同上有很多个这样的列表C=[[A,C1],[B,C2],[C,C3],[D,C4]] D=[[A,D1],[B,D2],[C,D3],[D,D4]]我要按照每个列表第一个元素进行合并最后生成结果是TMP = [[A,A1,B1,C1,D1],[B,A2,B2,C2,D2],[C,A3,B3,C3,D3]]最好有标准库不想做a*b*c*d次循环
或者改成字典合并dic1='A':1,'B':2,'C':3dic2='A':11,'B':22,'C':33合并成一个字典dic_all='A':[1,11],'B':[2,22],'C':[3,33]

---------------------------------------
对不起，有一个条件忘了加了。就是列表或者字典里面的字段不一定是排序排好的，而且也不一定A表里面和B表里面值是一样的
A=[['A','A1'],['B','A2'],['C','A3'],['D','A4']]
B=[['B','B2'],['A','B1'],['D','B4'],['C','B3']]
C=[['A','C1'],['B','C2'],['D','C4']]
D=[['B','D2'],['C','D3'],['D','D4']]

#!/usr/bin/env python
# coding: utf-8
#
# filename: lpart.py
# author: Tim Wang
# date: Dec., 2013

A=[['A','A1'],['B','A2'],['C','A3'],['D','A4']]
B=[['A','B1'],['B','B2'],['C','B3'],['D','B4']]
C=[['A','C1'],['B','C2'],['C','C3'],['D','C4']]
D=[['A','D1'],['B','D2'],['C','D3'],['D','D4']]

R = zip(zip(*A)[1], zip(*B)[1], zip(*C)[1], zip(*D)[1])
print R

dic1='A':1,'B':2,'C':3
dic2='A':11,'B':22,'C':33
dic3='A':111,'B':222,'C':333

keys = dic1.keys()
values = zip(
    zip(*dic1.items())[1],
    zip(*dic2.items())[1],
    zip(*dic3.items())[1]
)

dic_all=dict(zip(keys, values))
print dic_all

A=[['A','A1'],['B','A2'],['C','A3'],['D','A4']]
B=[['B','B2'],['A','B1'],['D','B4'],['C','B3']]
C=[['A','C1'],['B','C2'],['D','C4']]
D=[['B','D2'],['C','D3'],['D','D4']]

dicts = map(dict, (A,B,C,D))
totalkeys = reduce(lambda x,y: x|set(y.keys()), dicts, set())
dict_all = dict([(k, filter(None, map(lambda x: x.get(k), dicts))) 
    for k in totalkeys])

print dict_all

>python -u "lpart.py"
[('A1', 'B1', 'C1', 'D1'), ('A2', 'B2', 'C2', 'D2'), ('A3', 'B3', 'C3', 'D3'), ('A4', 'B4', 'C4', 'D4')]
'A': (1, 11, 111), 'C': (3, 33, 333), 'B': (2, 22, 222)
'A': ['A1', 'B1', 'C1'], 'C': ['A3', 'B3', 'D3'], 'B': ['A2', 'B2', 'C2', 'D2'], 'D': ['A4', 'B4', 'C4', 'D4']
>Exit code: 0    Time: 0.048

追问

列表第一个元素不是固定的而且顺序也不一定

追答

上面代码的第三个段落就是适应"列表第一个元素不是固定的而且顺序也不一定"情况的

参考技术A
A=[['A','A1'],['B','A2'],['C','A3'],['D','A4']]
B=[['A','B1'],['B','B2'],['C','B3'],['D','B4']]
C=[['A','C1'],['B','C2'],['C','C3'],['D','C4']]
D=[['A','D1'],['B','D2'],['C','D3'],['D','D4']]
arr=[A,B,C,D]
dic_all=x[0][0]:[y[1] for y in x] for x in zip(*arr)
list_all=[[x[0][0]]+[y[1]for y in x] for x in zip(*arr)]

if __name__=='__main__':
    print(list_all)
    print(dic_all)

 结果:

[['A', 'A1', 'B1', 'C1', 'D1'], ['B', 'A2', 'B2', 'C2', 'D2'], ['C', 'A3', 'B3', 'C3', 'D3'], ['D', 'A4', 'B4', 'C4', 'D4']]
'D': ['A4', 'B4', 'C4', 'D4'], 'A': ['A1', 'B1', 'C1', 'D1'], 'C': ['A3', 'B3', 'C3', 'D3'], 'B': ['A2', 'B2', 'C2', 'D2']

追问

列表第一个元素不是固定的而且顺序也不一定

参考技术B #!/usr/bin/env python
#-*- coding: utf-8 -*-
def merge(x, y):
for k,v in y.items():
try:
a_v = []
x_pre = x[k]
if isinstance(x_pre,list):
a_v.extend(x_pre)
else:
a_v.append(x_pre)
a_v.append(v)
x[k] = list(set(a_v))
except KeyError:
x[k] = v
return x

print reduce(merge, (dic1,dic2,dic3))
复杂度，O(n) , 对于列表可以先转换成字典！！本回答被提问者和网友采纳

Python list 两个不等长列表交叉合并

遇到一个需求，需要对两个长度不一定相等的列表进行交叉合并。像拉拉链一样（两边的拉链不一定相等）。

如：

　　a = [1, 3, 5]

　　b = [2, 4, 6, 8]

需将a, b 合并为 c

　　c = [1, 2, 3, 4, 6, 8]

网上又看到定义函数，或者使用 zip，自己觉得不太理想，就想倒腾下，当然有可能下述方法已经有人想到了。

方法一：　　写 for 循环

　　a = [1, 3, 5]

　　b = [2, 4, 6, 8]

　　c = [ ]

　　for i in range( max ( len( a ), len( b ) ) ):

　　　　if a:

　　　　　　c.append( a.pop() )

　　　　if b:

　　　　　　c.append( b.pop() )

方法二： 列表表达式（超级长，试了很久才搞出来）

　　a = [1, 3, 5]

　　b = [2, 4, 6, 8]

　　a.reverse()

　　b.reverse()

　　c = [ ( lambda i: a.pop() if ( a! = [ ]  and ( i % 2 ==0 or b==[ ] )) else b.pop() )( i ) for i in range( len( a ) + len( b ) )]　　

 

　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　-- by Clay