在Python中重塑不规则列表

Question

我想

wide_list = [[1,['a','b','c']],[2,['d','e']],[3,'f']]

以“长格式”重塑以下列表：

long_list = [[1,'a'],[1,'b'],[1,'c'],[2,'d'],[2,'e'],[3,'f']]

如何在Python中有效地实现这一点？

Answer 1

尝试嵌套列表理解：

>>> wide_list = [[1,['a','b','c']],[2,['d','e']],[3, ['f']]]
>>> long_list = [[k, v] for k, sublist in wide_list for v in sublist]
>>> long_list
[[1, 'a'], [1, 'b'], [1, 'c'], [2, 'd'], [2, 'e'], [3, 'f']]

请注意，必须更改最后一组以匹配前两组的模式。使用 [3, ['f']] 代替 [3, 'f']。否则，您将需要针对不遵循该模式的组的特殊情况逻辑。

Answer 2

实现此目的的一种方法是使用列表理解：

>>> [[x[0],letter] for x in wide_list for letter in x[1]]
[[1, 'a'], [1, 'b'], [1, 'c'], [2, 'd'], [2, 'e'], [3, 'f']]

Answer 3

使用 列表推导式，

long_list = [[x[0],item] for x in wide_list for item in x[1]]

但是 这个答案可能是最清楚的。