如何连接列表元素
这里我有列表列表,如果列表的第二个元素有名称,我只想连接前两个元素。别的什么都不做。
下面是我试过的代码:
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']]
我能找到任何方法来做到这一点吗?
回答
我不会想太多,只需连接元素然后添加其余元素
[[f"{i[0]}_{i[1]}" if len(i) == 4 else i[0], *i[-2:]] for i in lst_1]
- 非常简洁的解决方案