在python中使用列表理解擴展列表

Question

因此，讓我們說我有列表 項目 = ['abc'，'def'，'tre']。

現在我想在每個列表項前插入一個標誌。

例如，新列表項目2應該是['-g'，'abc'，'-g'，'def'，'-g'，'tre']。

我可以讀取列表並在追加標誌後追加每一個，但是我想開始以python的方式進行操作。

我想出是：

items2.extend(['-g', i] for i in items) 

這是什麼讓我有較小的列表清單：

items2 = [['-g', 'abc'], ['-g', 'def'], ['-g', 'tre']] 

我明白，這是因爲這將是等於說

items2.extend([ ... ] , [ ... ], [ ... ]) 

有關於此的任何想法？由於

Answer 1

[j for i in items for j in '-g', i] 

Answer 2

一種方法是使用zip：

>>> items = ['abc', 'def', 'tre'] 
>>> zip(['-g', '-g', '-g'], items) 
[('-g', 'abc'), ('-g', 'def'), ('-g', 'tre')] 

之後，您可以使用itertools.chain拼合列表：

>>> list(itertools.chain(*zip(['-g', '-g', '-g'], items))) 
['-g', 'abc', '-g', 'def', '-g', 'tre'] 

注意的是，由於函數調用的開銷，這是相當緩慢，與簡單的列表理解相比：

In [17]: %timeit [j for i in items for j in '-g', i] 
1000000 loops, best of 3: 923 ns per loop 

In [18]: %timeit list(itertools.chain(*zip(['-g', '-g', '-g'], items))) 
100000 loops, best of 3: 2.67 µs per loop 

因此對於performa我建議你使用Robᵩ answer。

Answer 3

爲補充其他優秀zip根據答案，你可能會使用itertools.repeat()代替「計數」項目：

>>> items = ['abc', 'def', 'tre'] 
>>> list(itertools.chain(*zip(itertools.repeat('-g'), items))) 
#        ^^^^^^^^^^^^^^^^^^^^^^ 
#        repeat as much as needed 
['-g', 'abc', '-g', 'def', '-g', 'tre'] 

如果你不喜歡拉鍊，一種替代方案：

>>> list(itertools.chain(*[('-g', i) for i in items])) 
['-g', 'abc', '-g', 'def', '-g', 'tre'] 

Answer 4

另一種簡單的方法是遍歷列表並在每個元素之後插入。

for i in range(len(items)): 
    items.insert((i*2)+1,'-g') 

當您插入長度可達插入點雙打，所以你需要把你的下一個元素在（I * 2）+1