列表理解结合了嵌套列表中的2个元素

如何解决列表理解结合了嵌套列表中的2个元素

是否可以使用列表理解功能根据字符的出现在嵌套列表中合并2个元素，例如：如果遇到“ +”与下一个元素合并？我有一些使用嵌套循环执行此操作的代码，但尝试使用列表理解来实现。

输入： l = [['A-2','A-3','A-4','+','100','A-5'],['B-2','B-3','B-4','500','B-5']]

输出： l = [['A-2','+100','+500','B-5']]

代码：

for nested in l: 
   z = iter(nested)
   for i in z:
      if i == '+':
         i = i+next(z)

解决方法

以下将起作用：

[[x + next(i) if x == "+" else x for x in i] for i in map(iter,l)]

# [['A-2','A-3','A-4','+100','A-5'],['B-2','B-3','B-4','+500','B-5']]

如果最后一个元素可能是"+"，则可以将默认值传递给next

next(i,"")

避免出错。