根据条件突出显示行

Question

我有一个包含两列日期的随机数据框，我一直在尝试突出显示条件日期列表中存在开始日期的行。这是我失败的尝试：

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)

因为我正在尝试使用 xlswriter 导出此数据框，所以我正在寻找一种即使在 excel 文件中也能保持背景颜色的方法。谢谢！

Answer 1

我们可以使用 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

或 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

在Excel中：

在 Jupyter Notebook 中

根据条件突出显示行

Highlighting rows based on a condition

conditional-formatting

pandas

xlsxwriter

pandas-styles