如何按索引連接列表中的元素？

Question

我的列表：

['0.125', '0', 'A'] 
['0.5', '0', 'B', ' A'] 
['0.75', '0', 'C'] 
['0.5', '0.5', 'D', ' C', 'A'] 

我想

['0.125', '0', 'A'] 
['0.5', '0', 'B'] 
['0.5', '0', 'A'] 
['0.75', '0', 'C'] 
['0.5', '0.5', 'D'] 
['0.5', '0.5', 'C'] 
['0.5', '0.5', 'A'] 

這樣做的任何指南？

Answer 1

您可以使用嵌套列表理解：

[sub[:2] + [elem] for sub in yourlist for elem in sub[2:]] 

演示：

>>> yourlist = [ 
...  ['0.125', '0', 'A'], 
...  ['0.5', '0', 'B', ' A'], 
...  ['0.75', '0', 'C'], 
...  ['0.5', '0.5', 'D', ' C', 'A'], 
... ] 
>>> from pprint import pprint 
>>> [sub[:2] + [elem] for sub in yourlist for elem in sub[2:]] 
[['0.125', '0', 'A'], ['0.5', '0', 'B'], ['0.5', '0', ' A'], ['0.75', '0', 'C'], ['0.5', '0.5', 'D'], ['0.5', '0.5', ' C'], ['0.5', '0.5', 'A']] 
>>> pprint(_) 
[['0.125', '0', 'A'], 
['0.5', '0', 'B'], 
['0.5', '0', ' A'], 
['0.75', '0', 'C'], 
['0.5', '0.5', 'D'], 
['0.5', '0.5', ' C'], 
['0.5', '0.5', 'A']]