通过 Pandas 保持 Excel 下拉列表

Question

我有一个 excel 文件，其中标题有下拉列表，可用于根据特定的列值（正是 WHERE 语句所做的）选择行。 我将此文件导入熊猫并执行一些操作。 假设我根据“emp_id”列删除重复值”

data = data.drop_duplicates(['emp_id'])

然后我将此数据框保存到excel，

data.to_excel("new_data.xlsx")

但是，此新数据在标题上没有任何下拉列表。 有没有办法保留下拉菜单或 python/pandas 不支持它？

Answer 1

如果我理解正确的话，这可以通过XlsxWriter轻松完成：

import pandas as pd

df = pd.DataFrame({
    'Numbers': [1, 2, 3, 4, 5],
    'Letters': ['a', 'b', 'c', 'd', 'e']
})

with pd.ExcelWriter('new_data.xlsx', engine='xlsxwriter') as writer:
    df.to_excel(excel_writer=writer, sheet_name='Filter', index=False)

    worksheet = writer.sheets['Filter']

    # set up autofilter
    worksheet.autofilter(0, 0, len(df.index) - 1, len(df.columns) - 1)