时间戳

首页 > 解决方案 > 通过熊猫保持Excel下拉列表

python-3.x - 通过熊猫保持Excel下拉列表

问题描述

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

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

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

data.to_excel("new_data.xlsx")

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

标签： python-3.xpandasexcel-2013

解决方案

如果我理解正确的话，这可以很容易地用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)

推荐阅读