python - 在python中的匹配元素上合并两个多维列表

Question

我正在尝试在匹配元素上合并两个多维列表。例如：

list1 = [['1', '2' ,'A'], ['3','4','D']]
list2 = [['A','A1'],['B','B1'],['C','C1'],['D','D1'],['D','D2']]

我想要的输出是：

[['1', '2', 'A', 'A1'],['3', '4', 'D', 'D1'],['3', '4', 'D', 'D2']]

谁能帮我？谢谢你。

标签： pythonlistmergeconcatenation

Answer 1

out = [ b + a[1:] for a in list2 for b in list1 if b[-1] == a[0] ]
print( out )

输出：

[['1', '2', 'A', 'A1'], ['3', '4', 'D', 'D1'], ['3', '4', 'D', 'D2']]