如何将Python Dataframe返回至PowerShell？参数传递与结果回传问题

问题根源 #

• PowerShell通过捕获Python脚本的**标准输出(stdout)**获取返回结果，Python函数的return语句仅作为函数内部返回值，不会输出到stdout，因此PowerShell无法捕获。
• 直接传递多行SQL查询时，PowerShell会将换行、空格分割为多个参数，导致Python接收的查询语句不完整。

解决方案步骤 #

1. 修改Python脚本，输出结构化结果到标准输出 #

将查询得到的DataFrame转换为CSV或JSON格式并打印到stdout，同时修复参数接收逻辑，确保获取完整的SQL查询。注意补充导入sys模块（原代码遗漏）。

修改后的Python代码：

import sys
import pandas as pd

def read_snowflake():
    # 保留你的Snowflake连接逻辑，无需修改
    pass

def main(query):
    snf_conn = read_snowflake()
    try:
        df = pd.read_sql(query, snf_conn)
        # 将DataFrame转为无索引的CSV格式输出，便于PowerShell解析
        print(df.to_csv(index=False))
    except Exception as e:
        # 错误信息输出到stdout，同时返回非0状态码标记失败
        print(f"ERROR: {str(e)}")
        sys.exit(1)

if __name__ == "__main__":
    # 检查是否传入参数
    if len(sys.argv) < 2:
        print("ERROR: 未传入SQL查询参数")
        sys.exit(1)
    # 拼接所有参数，处理查询语句被拆分的情况
    full_query = ' '.join(sys.argv[1:])
    main(full_query)

2. 修改PowerShell代码，正确传递参数并解析结果 #

使用Here-String定义多行SQL，并用双引号包裹参数确保完整传递；捕获输出后转换为PowerShell可操作的对象，同时添加错误判断。

修改后的PowerShell代码：

# 用Here-String定义多行SQL，避免换行/空格导致的参数拆分
$qry = @"
SELECT COUNT(*)
FROM source
WHERE the_DATE = DATEADD(DAY ,-1, CURRENT_DATE)
"@

# 双引号包裹$qry，确保完整传递给Python
$rawOutput = python C:\User\Python\Small_Data_Retr.py "$qry"

# 通过退出码判断Python脚本是否执行成功
if ($LASTEXITCODE -ne 0) {
    Write-Error "Python脚本执行失败: $rawOutput"
    exit 1
}

# 将CSV格式的输出转换为PowerShell对象
$results = $rawOutput | ConvertFrom-Csv

# 查看最终结果
$results

关键说明 #

• 参数传递：Here-String(@" ... "@)能保留SQL的换行格式，双引号包裹参数可避免PowerShell拆分查询语句；Python端用' '.join(sys.argv[1:])拼接所有传入的参数，确保获取完整的SQL。
• 结果解析：CSV是PowerShell原生支持的结构化格式，用ConvertFrom-Csv可直接转为可操作的对象；若偏好JSON，可将Python代码中的df.to_csv改为df.to_json(orient='records')，PowerShell端用ConvertFrom-Json解析。
• 错误处理：Python脚本出错时返回非0状态码，PowerShell通过$LASTEXITCODE快速判断执行状态，同时捕获错误信息便于排查。

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