使用pandas xlswriter模块使用conditional_format格式化整行

时间:2018-01-30 17:27:35

标签: python pandas xlsxwriter

我正在使用pandas xlswriter模块创建一个excel报告。在相同的代码片段下面。

number_rows = len(df.index)
//df is dataframe

writer = pd.ExcelWriter("Report_1.xlsx",engine='xlsxwriter')
df.to_excel(writer,"report")

workbook = writer.book
worksheet = writer.sheets['report']

# Define range for the color formatting
color_range = "A2:F{}".format(number_rows+1)


format1 = workbook.add_format({'bg_color': '#FFC7CE',
                               'font_color': '#9C0006'})

worksheet.conditional_format(color_range, {'type': 'text',
                                           'criteria' : 'containing',
                                           'value': 'SUCCESS',
                                           'format': format1})

我想突出显示一行(' bg_c​​olor':'#FFC7CE',' font_color':'#9C0006')基于单元格值(= SUCCESS)。但是当我使用" conditional_format"它只适用于那个特定的细胞。如果“条件”与单元格值匹配,有没有办法将格式应用于整行?

1 个答案:

答案 0 :(得分:3)

我在下面提供了一个完全可重现的示例,该示例使用Excel (Link here)中的INDIRECT()函数。请注意,我已将范围设置为$A$1:$B$6,但如果您有更多列,则可以将其扩展到不同的范围。

import pandas as pd

df = pd.DataFrame({"Name": ['A', 'B', 'C', 'D', 'E'], 
"Status": ['SUCCESS', 'FAIL', 'SUCCESS', 'FAIL', 'FAIL']})

number_rows = len(df.index) + 1

writer = pd.ExcelWriter('Report_1.xlsx', engine='xlsxwriter')

df.to_excel(writer, sheet_name='Sheet1', index=False)

workbook  = writer.book
worksheet = writer.sheets['Sheet1']

format1 = workbook.add_format({'bg_color': '#FFC7CE',
                              'font_color': '#9C0006'})

worksheet.conditional_format("$A$1:$B$%d" % (number_rows),
                             {"type": "formula",
                              "criteria": '=INDIRECT("B"&ROW())="SUCCESS"',
                              "format": format1
                             }
)

workbook.close()

预期输出为excel:

Expected Output