条件加入列表

Question

我有一个Python列表：

data = ['Cost', '$', 4244, '$', 4090, '$', 3967]

我想要做的是将$与前面的元素合并，并获得一个新列表：

data = ['Cost', '$4244', '$4090', '$3967']

实现这一目标的最佳方法是什么？

Answer 1

您可以使用iterators：

data_iter = iter(data)
[e + str(next(data_iter)) if e == '$' else e for e in data_iter]

这利用了我们可以获得迭代的下一个元素的事实，即使在迭代迭代时也是如此。每次找到'$'字符串时，下一个元素都会被拉入并连接成一个字符串。

演示：

>>> data = ['Cost', '$', 4244, '$', 4090, '$', 3967]
>>> data_iter = iter(data)
>>> [e + str(next(data_iter)) if e == '$' else e for e in data_iter]
['Cost', '$4244', '$4090', '$3967']