python - 如何在 pandas python> 中获取数据框中三列的所有组合

Question

三列的所有可能组合

我无法使用 itertools.combinations 或 itertools.permutations 来做到这一点。

输入数据框:

a   b   c
1   101 1001
2   102 1002
3   103 1003

预期数据框:

a   b   c
1   101 1001
1   101 1002
1   101 1003
1   102 1001
1   102 1002
1   102 1003
1   103 1001
1   103 1002
1   103 1003
2   101 1001
2   101 1002
2   101 1003
2   102 1001
2   102 1002
2   102 1003
2   103 1001
2   103 1002
2   103 1003
3   101 1001
3   101 1002
3   101 1003
3   102 1001
3   102 1002
3   102 1003
3   103 1001
3   103 1002
3   103 1003

Answer 1

使用itertools.product:

from  itertools import product
#all columns
df = pd.DataFrame(list(product(*df.values.T)))
#if you need to specify columns
#df = pd.DataFrame(list(product(*[df.a, df.b, df.c])))
print (df)