combining groups of columns with boolean values to create multiple new columns

Question

I was wondering how I can create new columns based on old ones by multiplying two groups of them in my dataframe. So if I have [A, B] and [X, Y]:

   A     B      X     Y
0  True  False  True  False

Now I want this to result in 4 extra columns, of which only the one that has both true is true:

   A     B      X     Y      AX     BX      AY     BY
0  True  False  True  False  True   False   False  False

In my situation, the groups are bigger, so I'm looking for a solution that can be used by using the two lists of column headers [A, B] and [X, Y]. I have tried list comprehension but I cannot get it to work :-(

Shubham Sharma · Accepted Answer · 2020-08-25 13:40:51Z

3

Use itertools.product to get the cartesian product of column names then use Series.mul inside a list comprehension to create corresponding column products, finally use pd.concat to concat these products with df:

from itertools import product

l1, l2 = ['A', 'B'], ['X', 'Y']

c = [df[a].mul(df[b]).rename(''.join([a, b])) for a, b in product(l1, l2)]
df = pd.concat([df] + c, axis=1)

Result:

      A      B     X      Y    AX     AY     BX     BY
0  True  False  True  False  True  False  False  False

edited Aug 25, 2020 at 13:40

answered Aug 25, 2020 at 13:39

Shubham Sharma

71.8k6 gold badges26 silver badges58 bronze badges

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

2 Comments

Charles Over a year ago

Amazing how fast you came up with that. Thank you! I could use this quite often!

Shubham Sharma Over a year ago

Happy coding :) @Charles

Collectives™ on Stack Overflow

combining groups of columns with boolean values to create multiple new columns

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related