Fetchall()转DataFrame时未获取Snowflake SQL列名问题求助

fetchall()方法只会返回查询结果的行数据元组，不会携带列名元数据，pandas接收无列名的二维数据生成DataFrame时，会默认使用从0开始的数字作为列名，也就是你输出中看到的0列。
另外你贴的原始代码存在几处会直接导致运行失败的语法问题：

• Snowflake连接初始化需要调用connect()方法，不能直接调用snowflake.connector
• SQL语句赋值缺少等号，存在语法错误
• 变量名大小写不一致，定义时用大写开头Ctx，后续调用时用小写ctx会触发变量不存在的报错

方案1：手动提取游标元数据映射列名 #

游标对象的description属性存储了查询返回列的全量元信息，元组第一个元素就是对应列名，可以提取后传入DataFrame构造参数。修正后代码如下：

import snowflake.connector
import pandas as pd

# 初始化连接，补全自身环境对应的账号、密码、仓库、库、schema参数
ctx = snowflake.connector.connect(
    user='users',
    password='your_account_password',
    account='your_snowflake_account_id',
    role='user_role',
    warehouse='your_compute_warehouse',
    database='your_target_database',
    schema='your_target_schema'
)

query = "SELECT product_id FROM product"
cursor = ctx.cursor()
resultset = cursor.execute(query)

# 提取列名传入DataFrame
df = pd.DataFrame(
    resultset.fetchall(),
    columns=[col[0] for col in cursor.description]
)

# 用完释放连接资源
cursor.close()
ctx.close()

方案2：使用连接器原生pandas读取方法（推荐） #

Snowflake Python连接器自带直接输出带列名pandas DataFrame的方法fetch_pandas_all()，不需要手动处理列名映射，代码更简洁：

import snowflake.connector
import pandas as pd

ctx = snowflake.connector.connect(
    # 同方案1，补全自身环境连接参数
    user='users',
    password='your_account_password',
    account='your_snowflake_account_id',
    role='user_role',
    warehouse='your_compute_warehouse',
    database='your_target_database',
    schema='your_target_schema'
)

query = "SELECT product_id FROM product"
cursor = ctx.cursor()
# 直接获取列名正确的DataFrame
df = cursor.execute(query).fetch_pandas_all()

cursor.close()
ctx.close()

注意：使用fetch_pandas_all()需要提前安装对应依赖，执行命令pip install "snowflake-connector-python[pandas]"即可完成依赖安装。

两种方案执行后，再运行print(df)都会正确显示product_id作为列名，不会再出现默认数字列名的问题。

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