python - 如何使用 pandas 在标题行中应用过滤器

Question

假设我有这样的数据框:

     col1    col2          col3
0    A        A_1          pass
1    A        A_2          pass
2    A        A_1          fail
3    A        A_1          fail
4    A        A_1          pass
5    A        A_2          fail

我想将此数据框保存到 Excel 文件中，并在标题“col1”“col2”“col3”中应用过滤器，因此当我打开 Excel 文件时，过滤器应该已经应用在标题中。

我使用此行将其保存为 Excel 文件，但未应用过滤器:

df.to_excel('data.xlsx',index=False)

我该怎么做？

Answer 1

试试这个:

import pandas as pd
from openpyxl import load_workbook

path = r'D:\temp\you.xlsx'
df.to_excel(path,index=False)

wb = load_workbook(path)
ws = wb.active
ws.auto_filter.ref = ws.dimensions
wb.save(path)
wb.close()