Pandas.style.apply .to_excel выдает ошибку ключа ⇐ Python

[Anonymous]

У меня есть df с тремя столбцами A, B, C, и я хочу раскрасить ячейку B, если A и B< /code> равны. Я также хочу экспортировать его, чтобы улучшить стиль.

Код: Выделить всё

data = [[10, 30, 30], [15, 20, 33], [14, 40, 36]]
df = pd.DataFrame(data, columns=['A', 'B', 'C'])

def highlight(df, color):
return np.where(df['A'] == df['B'], f"color: {color};", None)
df.style.apply(highlight, color='red', subset=['B']).to_excel(os.path.join(".\\data", 'highlighted_data.xlsx'), index=False)

Но я получаю ошибку такого вида:

Код: Выделить всё

---------------------------------------------------------------------------
KeyError                                  Traceback (most recent call last)
Cell In[4], line 7
5 def highlight_max(df, color):
6     return np.where(df['A'] == df['B'], f"color: {color};", None)
----> 7 df.style.apply(highlight_max, color='red', subset=['A']).to_excel(os.path.join(".\\data", 'highlighted_data.xlsx'), index=False)
8 # df
9 # ------------   ------------
11 def highlight_pred(df, output_dir):

File d:\Apps\envs\cuda11\Lib\site-packages\pandas\io\formats\style.py:537, in Styler.to_excel(self, excel_writer, sheet_name, na_rep, float_format, columns, header, index, index_label, startrow, startcol, engine, merge_cells, encoding, inf_rep, verbose, freeze_panes, storage_options)
509 @doc(
510     NDFrame.to_excel,
511     klass="Styler",
(...)
533     storage_options: StorageOptions | None = None,
534 ) -> None:
535     from pandas.io.formats.excel import ExcelFormatter
--> 537     formatter = ExcelFormatter(
538         self,
539         na_rep=na_rep,
540         cols=columns,
541         header=header,
542         float_format=float_format,
543         index=index,
...
--> 417     raise KeyError(key)
418 self._check_indexing_error(key)
419 raise KeyError(key)

KeyError: 'A'

И я понятия не имею, что происходит.
Я пытался установить axis=1

Код: Выделить всё

df.style.apply(highlight, color='red', subset=['A'], axis=1).to_excel(os.path.join(".\\data", 'highlighted_data.xlsx'), index=False)

но безуспешно.

Подробнее здесь: https://stackoverflow.com/questions/790 ... g-keyerror