python-3.x - 通过 pandas 保留 Excel 下拉列表

Question

我有一个 Excel 文件，其中标题有下拉菜单，可用于根据特定列值选择行(正是 WHERE 语句的作用)。我将此文件导入 pandas 并执行一些操作。假设我根据“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)