Передайте переменную в xlsxwriter для pandas условного форматирования - PullRequest
Новая
       5

Передайте переменную в xlsxwriter для pandas условного форматирования

0 голосов
/ 07 февраля 2020

Я выполняю for l oop в кадре данных и хочу выделить строку в выходном отчете, когда она соответствует определенным условиям. Тем не менее, мой код работает для жестко закодированного значения (то есть 10), но я хочу передать переменную. Любая помощь приветствуется. 

df = pd.DataFrame({"ID": [10, 11, 12, 13, 14], 
"NAME": ['item1', 'item2', 'item3', 'item4', 'item5']})

number_rows = len(df.index) + 1
writer = pd.ExcelWriter("Report.xlsx",engine='xlsxwriter')

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

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


a = df['ID'][1]

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

worksheet.conditional_format("$A$1:$B$%d" % (number_rows),
                             {"type": "formula",
                              "criteria": '=INDIRECT("A"&ROW())= 10',  # I want to pass variable (a) instead of '10'
                              "format": format1
                             }
)
workbook.close()

1 Ответ

0 голосов
/ 07 февраля 2020

Найден способ добавления переменной с использованием атрибута "Значение" 

df = pd.DataFrame({"ID": [10, 11, 12, 13, 14], 
"Status": ['item1', 'item2', 'item3', 'item4', 'item5']})

number_rows = len(df.index) + 1
writer = pd.ExcelWriter("Report.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": "cell",
                              'criteria': '=',
                              "value": a,           #here a is variable now
                              "format": format1
                             }
)
workbook.close()
...