使用python使用下拉菜单时，如何在excel中更改整个行的颜色

Question

当使用 Python 和 XlsxWriter data_validation() 下拉具有两个值 High 和 Low 时如何更改整个单元格的背景？

如果使用下拉选择“高”，则整行必须为绿色。如果使用下拉选择低，则整行必须为红色。

我在我的代码中使用了以下几行，或者我尝试过这样的

workbook = xlsxwriter.Workbook('Reports_Sample.xlsx') ##工作表标题

sheet1 = workbook.add_worksheet("Report1")     ## Sheet 1

heading=["A","B","C","D"]

## Now i am adding drop down as High and Low to the column D

sheet1.data_validation('D2:D7', {'validate': 'list',
                                 'source': ['High', 'Low']})   

如果点击高，该行应该是绿色的。如果单击“低”，则该行应为红色。

我无法为其添加背景颜色。

Answer 1

99% 的事情您可以使用 excel 手动完成，您可以使用强大的库 xlsxwriter 将它们自动化。

话虽如此，我会这样做：

import pandas as pd
import numpy as np

# Create a test df
df = pd.DataFrame({'A': [1,2,3,4],
                    'B': ['a','b','c','d'],
                    'C': [2.2,np.nan,8.1,9.0],
                    'D': [np.nan,np.nan,np.nan,np.nan]})

# Kickstart the xlsxwriter
writer = pd.ExcelWriter('test.xlsx', engine='xlsxwriter')
df.to_excel(writer, sheet_name='Sheet1', index=False)
workbook  = writer.book
worksheet = writer.sheets['Sheet1']

# Define the formats
format_green = workbook.add_format({'bg_color': 'green'})
format_red = workbook.add_format({'bg_color': 'red'})

# Apply the data validation rule
worksheet.data_validation('D2:D5', {'validate': 'list',
                            'source': ['High','Low']})

# Apply the conditional formating rules
worksheet.conditional_format('A2:D5', {'type': 'formula',
                            'criteria': '=$D2="High"',
                            'format': format_green})

worksheet.conditional_format('A2:D5', {'type': 'formula',
                            'criteria': '=$D2="Low"',
                            'format': format_red})

writer.save()

输出：