一尘不染

用另一个列表的内容替换列表项

python

此问题类似,但我不想将一个项目的任何出现都替换为列表的内容,而不是将另一个项目替换为另一个。

orig = [ 'a', 'b', 'c', 'd', 'c' ]
repl = [ 'x', 'y', 'z' ]
desired = [ 'a', 'b', 'x', 'y', 'z', 'd', 'x', 'y', 'z' ]

# these are all incorrect, or fail to compile
[ repl if x == 'c' else x for x in orig ]
[ [a for a in orig] if x == 'c' else x for x in orig ]
[ (a for a in orig) if x == 'c' else x for x in orig ]
[ a for a in orig if x == 'c' else x for x in orig ]

编辑:明确表示我打算替换 所有 出现的项目,而不仅仅是第一个。(对未在回答中涉及此情况的任何人表示歉意。)


阅读 151

收藏
2021-01-20

共1个答案

一尘不染

不同的方法:当我进行替换时,我更喜欢从字典的角度进行思考。所以我会做类似的事情

>>> orig = [ 'a', 'b', 'c', 'd' ]
>>> rep = {'c': ['x', 'y', 'z']}
>>> [i for c in orig for i in rep.get(c, [c])]
['a', 'b', 'x', 'y', 'z', 'd']

最后一行是标准拼合成语。

这种方法的一个优点(劣势?)是它将处理的多次出现'c'

[更新:]

或者,如果您愿意:

>>> from itertools import chain
>>> list(chain.from_iterable(rep.get(c, [c]) for c in orig))
['a', 'b', 'x', 'y', 'z', 'd']

关于修改后的测试用例:

>>> orig = [ 'a', 'b', 'c', 'd', 'c' ]
>>> rep = {'c': ['x', 'y', 'z']}
>>> list(chain.from_iterable(rep.get(c, [c]) for c in orig))
['a', 'b', 'x', 'y', 'z', 'd', 'x', 'y', 'z']
2021-01-20