來自給定類別的值的所有可能組合的數據幀

Question

我有

{"A":[0,1], "B":[4,5], "C":[0,1], "D":[0,1]}

我想要的是什么

A   B   C   D
0   4   0   0
0   4   0   1
0   4   1   0
0   4   1   1
1   4   0   1

...等等。 基本上每個類別的所有值組合。 實現這一目標的最佳方法是什么？

Answer 1

如果x是你的dict：

>>> pandas.DataFrame(list(itertools.product(*x.values())), columns=x.keys())
    A  C  B  D
0   0  0  4  0
1   0  0  4  1
2   0  0  5  0
3   0  0  5  1
4   0  1  4  0
5   0  1  4  1
6   0  1  5  0
7   0  1  5  1
8   1  0  4  0
9   1  0  4  1
10  1  0  5  0
11  1  0  5  1
12  1  1  4  0
13  1  1  4  1
14  1  1  5  0
15  1  1  5  1

如果您希望按特定順序排列列，則需要在之后切換它們（例如， df[["A", "B", "C", "D"]] 。