搜索列表元素，并通过多个项目替换它

Question

什么是Python的方式通过给定列表['a', 'b', 'c']搜索元素b替换和插入多个项目b1, b2, b3以便列表最后读为['a', 'b1', 'b2', 'b3', 'c']

Answer 1

使用切片表示法：

>>> lst = ['a', 'b', 'c'] 
>>> i = lst.index('b') # This raises ValueError if there's no 'b' in the list. 
>>> lst[i:i+1] = 'b1', 'b2', 'b3' 
>>> lst 
['a', 'b1', 'b2', 'b3', 'c'] 

注意这仅改变第一个匹配项。

Answer 2

另一种方法：使用itertools.chain.from_iterable

>>> b = ["b1", "b2", "b3"] 
>>> a = ['a', 'b', 'c'] 
>>> a = [b if x=='b' else [x] for x in a] 
>>> a 
['a', ['b1', 'b2', 'b3'], 'c'] 
>>> import itertools 
>>> list(itertools.chain.from_iterable(a)) 
['a', 'b1', 'b2', 'b3', 'c'] 
>>>