搜索列表元素，並通過多個項目替換它

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