python - 如何连接列表元素

Question

这里我有列表列表，如果列表的第二个元素有名称，我只想连接前两个元素。否则什么都不做。

以下是我尝试过的代码:

lst_1 = [['ANTWERP' 'BRIDGE', '05', 'N'], 
         ['NORTHERN' 'VIGOUR', '05', 'N'],
         ['BRIDGE', '98', 'N']]
for i in lst_1:
    for j in i:               
        j[0:2] = ['_'.join(j[0:2])]

预期输出:

[['ANTWERP_BRIDGE', '05', 'N'], 
 ['NORTHERN_VIGOUR', '05', 'N'],
 ['BRIDGE', '98', 'N']]

我能找到任何方法吗？

Answer 1

我不会想太多，只需连接元素然后添加其余元素

 [[f"{i[0]}_{i[1]}" if len(i) == 4 else i[0], *i[-2:]] for i in lst_1]