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

Question

与此问题类似，但我不想用一个项目替换另一个项目，而是想用列表的内容替换任何一个项目。

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 ]

编辑：说清楚我打算替换所有出现的项目，而不仅仅是第一次。 （向在答案中未涉及该案件的任何人道歉。）

Answer 1

>>> orig = [ 'a', 'b', 'c', 'd' ]
>>> repl = [ 'x', 'y', 'z' ]
>>> desired = list(orig)  #can skip this and just use `orig` if you don't mind modifying it (and it is a list already)
>>> desired[2:3] = repl
>>> desired
['a', 'b', 'x', 'y', 'z', 'd']

当然，如果您不知道'c'在索引2处，您可以使用orig.index('c')来查找该信息。

Answer 2

不同的方法：当我做替换时，我更喜欢用字典来思考。 所以我会做类似的事情

>>> 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']

Answer 3

不需要任何花哨的东西：

desired = orig[:2] + repl + orig[3:]

要查找2您可以搜索orig.index('c') 。

x = orig.index('c')
desired = orig[:x] + repl + orig[x+1:]

如果repl不是列表，只需使用list(repl)

Answer 4

如果向后枚举，则可以随时扩展列表，因为您移动的项目已经完成了枚举。

>>> orig = [ 'a', 'b', 'c', 'd', 'c' ]
>>> repl = [ 'x', 'y', 'z' ]
>>> desired = [ 'a', 'b', 'x', 'y', 'z', 'd', 'x', 'y', 'z' ]
>>> for i in xrange(len(orig)-1, -1, -1):
...     if orig[i] == 'c':
...             orig[i:i+1] = repl
... 
>>> orig
['a', 'b', 'x', 'y', 'z', 'd', 'x', 'y', 'z']

Answer 5

另一种方式：

>>> import operator
>>> orig = [ 'a', 'b', 'c', 'd', 'c' ]
>>> repl = [ 'x', 'y', 'z' ]
>>> output = [repl if x == 'c' else [x] for x in orig]
>>> reduce(operator.add, output)
['a', 'b', 'x', 'y', 'z', 'd', 'x', 'y', 'z']
>>>