如何从包含列表和字符串的嵌套列表中找到所有可能的组合？

我试图从列表中获取所有可能的模式，例如：

input_x = ['1', ['2', '2x'], '3', '4', ['5', '5x']]

正如我们所看到的，它有 2 个嵌套列表['2', '2x'] and ['5', '5x'] here.

这意味着所有可能的模式为 4（2 个案例 x 2 个案例），预期输出为：

output1 = ['1','2' , '3', '4',  '5']
output2 = ['1','2x', '3', '4',  '5']
output3 = ['1','2' , '3', '4', '5x']
output4 = ['1','2x', '3', '4', '5x']

我尝试搜索如何搜索，但找不到任何示例（因为我不知道要搜索的“关键字”）

我认为 python 有内部库/方法来处理它。

实现此目的的一种方法是通过使用itertools.product。但要使用它，您需要首先将列表中的单个元素包装到另一个列表。

例如，首先我们需要转换您的列表：

['1', ['2', '2x'], '3', '4', ['5', '5x']]

to:

[['1'], ['2', '2x'], ['3'], ['4'], ['5', '5x']]

这可以通过以下列表理解来完成：

formatted_list = [(l if isinstance(l, list) else [l]) for l in my_list]
# Here `formatted_list` is containing the elements in your desired format, i.e.:
#    [['1'], ['2', '2x'], ['3'], ['4'], ['5', '5x']]

现在打电话itertools.product在上面的解压版本上list:

>>> from itertools import product

#                v  `*` is used to unpack the `formatted_list` list
>>> list(product(*formatted_list))
[('1', '2', '3', '4', '5'), ('1', '2', '3', '4', '5x'), ('1', '2x', '3', '4', '5'), ('1', '2x', '3', '4', '5x')]