Сохранение раскрывающегося списка Excel через pandas - PullRequest
Новая
       41

Сохранение раскрывающегося списка Excel через pandas

1 голос
/ 15 января 2020

У меня есть файл Excel, в котором заголовки имеют выпадающие списки, которые можно использовать для выбора строк на основе заданных значений c столбцов (именно то, что делает оператор WHERE). Я импортирую этот файл в pandas и выполняю некоторые операции. Допустим, я отбрасываю повторяющиеся значения на основе столбца «emp_id»: 

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

Затем я сохраняю этот фрейм данных в excel, 

data.to_excel("new_data.xlsx")

Однако эти новые данные не удаляются. вниз заголовка. Есть ли способ сохранить выпадающий или Python / pandas не поддерживает его?

1 Ответ

3 голосов
/ 15 января 2020

Если я вас правильно понимаю, это можно сделать довольно просто с помощью 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)
...