python - 如何在输出前自动过滤Excel工作表

我正在为从API获取数据的脚本编写函数，该API会将数据帧输出到Excel工作表。我需要自动过滤（作为可选参数）。

我快完成了，但是继续收到属性错误，并带有以下说明'Worksheet'对象没有属性'autofilter'

这是我一直在尝试的代码

def generate_xls(df, file_name, sheet_name, column_names=None, auto_filter=True):

    if column_names != None:
        df.columns = column_names

    print("appending the data in order...")

    print("\nCreating excel-file")

    writer = pd.ExcelWriter(file_name + '.xls')

    df.to_excel(writer,sheet_name,index=False)

    if auto_filter == True:
        for worksheet in writer.sheets.values():
            worksheet.autofilter(df.columns)

    writer.save()
    print("Excel-file generated.")

无需自动过滤，它就可以正常工作，但是对我来说，拥有它确实很重要，因为还有其他人可能需要它。任何建议都很好！

最佳答案

感谢您的输入。我在writer对象中使用pandas和xlsxwriter引擎解决了该问题：

    if auto_filter == True:
        for worksheet in writer.sheets.values():
            row_count = len(df.index)
            column_count = len(df.columns)
            worksheet.autofilter(0, 0, row_count-1, column_count-1)
    writer.save()
    print("Excel-file generated.")

关于python - 如何在输出前自动过滤Excel工作表，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/58100181/