Snowpark中union_all返回空值问题排查

两个关键问题 #

• show()方法返回None：你把df1.union_all(df2).show()赋值给了df3，但show()仅用于在控制台打印DataFrame内容，它的返回值是None，这就导致df3实际是空值，自然返回结果为空。
• DataFrame列结构不匹配：df1的列名是a、b，df2是c、d，虽然union_all会按位置强行合并，但结果列名会沿用df1的命名，这大概率不符合你的预期，建议先对齐列结构。

修正后的代码 #

如果希望两个DataFrame按同列名合并：

import snowflake.snowpark as snowpark

def main(session: snowpark.Session): 

    df1 = session.create_dataframe([[1, 2], [3, 4]], schema=["a", "b"])
    # 直接让df2使用和df1一致的列名
    df2 = session.create_dataframe([[0, 1], [3, 4]], schema=["a", "b"])
    # 先获取合并后的DataFrame，再单独调用show()打印
    df3 = df1.union_all(df2)
    df3.show()
    return df3

如果需要保留df2原列名再合并，可先重命名列：

import snowflake.snowpark as snowpark

def main(session: snowpark.Session): 

    df1 = session.create_dataframe([[1, 2], [3, 4]], schema=["a", "b"])
    df2 = session.create_dataframe([[0, 1], [3, 4]], schema=["c", "d"])
    # 重命名df2的列以匹配df1
    df2_renamed = df2.with_columns_renamed({"c": "a", "d": "b"})
    df3 = df1.union_all(df2_renamed)
    df3.show()
    return df3

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