How to get all combinations of three columns in a dataframe in pandas python>

Question

All possible combinations of three columns

I am not able to do it with itertools.combinations or itertools.permutations.

Input dataframe :

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

Expected dataframe :

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

niemand · Accepted Answer · 2021-03-16 12:57:36Z

4

Use 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)

edited Mar 16, 2021 at 12:57

niemand

475 bronze badges

answered Aug 30, 2019 at 10:57

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How to get all combinations of three columns in a dataframe in pandas python>

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related