CODEWITHSUNDEEP
pandasconditional-formattingxlsxwriterpandas-styles
Keyser Soze
Keyser Soze

Reputation: 272

Highlighting rows based on a condition

I have this random dataframe containing two columns with dates, I've been trying to highlight rows where the start date exists inside a list of condition dates. Here goes my failed attempt:

import pandas as pd 
import numpy as np
import datetime

df = pd.DataFrame({"Start": pd.date_range("1-jan-2021", periods=10, freq="1H")}).assign(**{"End": lambda d: d["Start"]+pd.Timedelta(hours=20)})

date_condition = ['2021-01-01 05:00:00','2021-01-01 08:00:00', '2021-01-01 02:00:00']
df = df.style.applymap(lambda x: 'background-color : yellow' if x['Start'] in date_condition)

Since I'm trying to export this dataframe using xlswriter, I'm looking for a method to keep the background color even in the excel file. Thank you !

Upvotes: 5

Views: 4645

Answers (1)

Henry Ecker
Henry Ecker

Reputation: 35626

We can use a combination of np.where and Series.isin:

date_condition = ['2021-01-01 05:00:00', '2021-01-01 08:00:00',
                  '2021-01-01 02:00:00']

highlighted_rows = np.where(df['Start'].isin(date_condition),
                            'background-color: yellow',
                            '')
# Apply calculated styles to each column:
styler = df.style.apply(lambda _: highlighted_rows)
styler.to_excel('output.xlsx')  # use styler to export to excel
# styler  # to display in NoteBook

Or with isin + Series.map:

date_condition = ['2021-01-01 05:00:00', '2021-01-01 08:00:00',
                  '2021-01-01 02:00:00']

highlighted_rows = df['Start'].isin(date_condition).map({
    True: 'background-color: yellow',
    False: ''
})
# Apply calculated styles to each column:
styler = df.style.apply(lambda _: highlighted_rows)
styler.to_excel('output.xlsx')  # use styler to export to excel
# styler  # to display in NoteBook

In Excel:

styled spreadsheet

In Jupyter Notebook

styled table

Upvotes: 9

Related Questions