Combine values in pandas dataframe to string

Question

I have a dataframe similar to this:

       Male  Over18  Single
0     0       0       1
1     1       1       1
2     0       0       1

I would like an extra column which gets a commaseperated string with the columnnames where the value is 1:

   Male  Over18  Single        CombinedString
0     0       0       1                Single
1     1       1       1  Male, Over18, Single
2     0       0       1                Single

Hope there is someone out there who can help :)

cs95 · Accepted Answer · 2021-05-01 20:39:15Z

7

One pandaic way is to perform a pandas dot product with the column headers:

df['CombinedString'] = df.dot(df.columns+',').str.rstrip(',')
df

   Male  Over18  Single      CombinedString
0     0       0       1              Single
1     1       1       1  Male,Over18,Single
2     0       0       1              Single

answered May 1, 2021 at 20:39

cs95

406k106 gold badges744 silver badges797 bronze badges

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

Comments

Umar.H · Accepted Answer · 2021-05-01 20:48:53Z

1

Another method would be to use .stack() and groupby.agg()

df['CombinedString'] = df.mask(df.eq(0)).stack().reset_index(1)\
                     .groupby(level=0)['level_1'].agg(','.join)


print(df)

   Male  Over18  Single      CombinedString
0     0       0       1              Single
1     1       1       1  Male,Over18,Single
2     0       0       1              Single

answered May 1, 2021 at 20:48

Umar.H

23.1k7 gold badges50 silver badges94 bronze badges

Collectives™ on Stack Overflow

Combine values in pandas dataframe to string

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related