Dash数据表下载到excel

import dash
import dash_html_components as html
import dash_table
import pandas as pd
from dash.dependencies import Output, Input
import dash_core_components as dcc
import base64

# 创建Dash应用程序
app = dash.Dash(__name__)

# 创建示例数据表
data = {'Name': ['John', 'Mike', 'Sarah'],
        'Age': [25, 30, 35],
        'City': ['New York', 'London', 'Paris']}
df = pd.DataFrame(data)

# 创建数据表组件
table = dash_table.DataTable(
    id='table',
    columns=[{"name": i, "id": i} for i in df.columns],
    data=df.to_dict('records')
)

# 创建下载按钮
download_button = html.A(
    'Download Excel',
    id='download-button',
    download="data.xlsx",
    href="",
    target="_blank"
)

# 布局
app.layout = html.Div([
    table,
    download_button
])

# 回调函数，处理下载按钮点击事件
@app.callback(
    Output('download-button', 'href'),
    [Input('table', 'data')]
)
def update_download_link(data):
    df = pd.DataFrame(data)
    # 生成Excel文件
    excel_file = df.to_excel(index=False, encoding='utf-8')
    # 将Excel文件转换为base64编码的字符串
    excel_data = base64.b64encode(excel_file).decode('utf-8')
    # 返回下载链接
    return f'data:application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;base64,{excel_data}'

if __name__ == '__main__':
    app.run_server(debug=True)