如何用Pandas与XlsxWriter将Excel文件存储到Snowflake内部Stage？

问题原因 #

Pandas 的 to_excel 仅支持本地文件系统路径，无法直接识别 Snowflake 的 Stage 路径格式（如 @JV_TEST），因此会抛出目录不存在的错误。要保留 Pandas 和 XlsxWriter 的操作，需分两步完成：先在本地临时目录生成 Excel 文件，再将文件上传到 Snowflake 内部 Stage。

修改后的代码 #

import snowflake.snowpark as snowpark
import xlsxwriter
import pandas as pd
from snowflake.snowpark.functions import col
import os

def main(session: snowpark.Session): 
    tableName = 'information_schema.packages'
    dataframe = session.table(tableName).filter(col("language") == 'python')

    # 转换为Pandas DataFrame
    x = dataframe.to_pandas()
    
    # 第一步：写入本地临时文件（Snowflake支持/tmp作为临时目录）
    local_file_path = "/tmp/jv.xlsx"
    # 显式指定XlsxWriter作为引擎生成Excel，可在此添加格式自定义操作
    with pd.ExcelWriter(local_file_path, engine='xlsxwriter') as writer:
        x.to_excel(writer, index=False)
    
    # 第二步：将本地文件上传到Snowflake内部Stage
    stage_location = "@JV_TEST"
    # auto_compress=False保留原始Excel格式，overwrite=True覆盖同名文件
    session.file.put(local_file_path, stage_location, auto_compress=False, overwrite=True)
    
    # 清理临时文件（可选）
    os.remove(local_file_path)
    
    dataframe.show()
    return dataframe

关键说明 #

• 临时文件路径：Snowflake 的 Python 运行环境中，/tmp 是可读写的临时目录，适合存放生成的 Excel 文件。
• Excel 自定义操作：在 pd.ExcelWriter 的上下文里，你可以通过 XlsxWriter 的 API 添加单元格样式、公式、图表等自定义操作，满足你的格式需求。
• 文件上传配置：session.file.put() 是 Snowpark 官方提供的文件上传方法，overwrite=True 可覆盖 Stage 中已存在的同名文件；若需保留原始 Excel 文件格式，需设置 auto_compress=False（默认压缩为 gz 格式）。
• 临时文件清理：上传完成后删除本地临时文件，避免占用运行环境资源。

内容的提问来源于stack exchange，提问作者jvels