I have highlighted pandas dataframe using conditional statement (highlight one column as yellow if other column value is True
). Before exporting to Excel I want to remove the conditional column I used. But I got error message.
Here is what I have done:
Here is my code:
def highlight_col(val):
data = pd.DataFrame('', index=val.index, columns=val.columns)
if val['Email Updated'].any():
data.loc[val['Email Updated'], 'Email'] = f"background-color: yellow"
return data
styled_df = df.style.apply(highlight_col, axis=None)
styled_df.to_excel('filename.xlsx', index = False)
After export I want to drop Email Updated
column and keep email
column.
So far I have tried
data.drop('Email Updated', axis = 1, inplace = True)
and
styled_df.drop('Email Updated', axis = 1, inplace = True)
But without success.
Thank you!