Question

我有一个元素列表，如：

x = ['A, B', '2', '3', 'Jan, Feb']
#      ^                   ^
#    comma separated single string

我想将此列表转换为列表列表，其中每个元素的所有组合都以逗号分隔，以便它们看起来像：

[['A', '2', '3', 'Jan'], ['A', '2', '3', 'Feb']
 ['B', '2', '3', 'Jan'], ['B', '2', '3', 'Feb']]

使用Python的内置函数/库实现这一目标的优雅方法是什么？

Answer 1

您可以先根据逗号,拆分列表元素，然后使用itertools.product获取所需的列表：

from itertools import product
x = ['A, B', '2', '3', 'Jan, Feb']

new_list = list(product(*[i.split(', ') for i in x]))

new_list的内容将是：

[('A', '2', '3', 'Jan'), 
 ('A', '2', '3', 'Feb'), 
 ('B', '2', '3', 'Jan'), 
 ('B', '2', '3', 'Feb')]

根据Dan D的评论，下面是Python 2中间列表V / s中间生成器比较的timeit比较。中间列表表达式略快：

mquadri$ python -m timeit "from itertools import product; list(product(*(i.split(', ') for i in ['A, B', '2', '3', 'Jan, Feb'])))"
100000 loops, best of 3: 4.83 usec per loop

mquadri$ python -m timeit "from itertools import product; list(product(*[i.split(', ') for i in ['A, B', '2', '3', 'Jan, Feb']]))"
100000 loops, best of 3: 3.78 usec per loop

Python - 将列表转换为所有组合列表的优雅方式

1 个答案: