pandas create new column based on boolean value

Question

I have a dataframe like this with Boolean values

black	yellow	orange
TRUE	TRUE	TRUE
FALSE	TRUE	FALSE
TRUE	TRUE	FALSE
FALSE	FALSE	TRUE

I want a separate column that summarizes the Boolean values based on column name which the column would be

summary
black, yellow, orange
yellow
black, yellow
orange

Any idea how to do this please? Thanks!

Code Different · Accepted Answer · 2022-12-29 02:03:46Z

1

You can use each row as a selection mask to filter the column names:

(
    df.astype("bool")
    .apply(lambda row: ", ".join(df.columns[row]), axis=1)
    .to_frame("summary")
)

answered Dec 29, 2022 at 2:03

Code Different

93.4k16 gold badges154 silver badges175 bronze badges

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

Comments

Scott Boston · Accepted Answer · 2022-12-29 02:38:49Z

1

Try this using pd.DataFrame.dot:

df_colors['summary'] = df_colors.dot(df_colors.columns+', ').str.strip(', ')
df_colors

Output:

   black  yellow  orange                summary
0   True    True    True  black, yellow, orange
1  False    True   False                 yellow
2   True    True   False          black, yellow
3  False   False    True                 orange

answered Dec 29, 2022 at 2:38

Scott Boston

154k15 gold badges160 silver badges207 bronze badges

Collectives™ on Stack Overflow

pandas create new column based on boolean value

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