python - 从列表中创建 Pandas 排列的数据框

Question

我有以下列表:

aa = ['aa1', 'aa2', 'aa3', 'aa4', 'aa5']
bb = ['bb1', 'bb2', 'bb3', 'bb4', 'bb5']
cc = ['cc1', 'cc2', 'cc3', 'cc4', 'cc5']

我想这样创建一个 pandas 数据框:

aa    bb    cc
aa1   bb1   cc1
aa2   bb1   cc1
aa3   bb1   cc1
aa4   bb1   cc1
aa5   bb1   cc1
aa1   bb2   cc1
aa1   bb3   cc1
aa1   bb4   cc1
aa1   bb5   cc1
aa1   bb1   cc2
aa1   bb1   cc3
aa1   bb1   cc4
aa1   bb1   cc5

我不知道该怎么做。 我看过例子: How to generate all permutations of a list in Python

我可以使用以下方法单独进行每个排列:

import itertools
itertools.permutations(['aa1','aa2','aa3','aa4','aa5'])

我有几十个列表，理想情况下，我想自动完成它们。

感谢任何帮助!

Answer 1

我相信您需要 itertools.product，而不是permutations。

In [287]: lists = [aa, bb, cc]

In [288]: pd.DataFrame(list(itertools.product(*lists)), columns=['aa', 'bb', 'cc'])
Out[288]: 
      aa   bb   cc
0    aa1  bb1  cc1
1    aa1  bb1  cc2
2    aa1  bb1  cc3
3    aa1  bb1  cc4
4    aa1  bb1  cc5
5    aa1  bb2  cc1
6    aa1  bb2  cc2
7    aa1  bb2  cc3
8    aa1  bb2  cc4
...

这将为您提供列表的笛卡尔积。截至目前，列名是硬编码的，但您可以使用 df.rename 动态重命名它们。