Skip to main content
 首页 » 编程设计

python之使用 SQL 查询结果列名作为 pandas dataframe 列名

2025年05月04日14itcoder

如何将 sql 查询中的列名添加到 pandas 数据框。我正在执行以下操作,但 columns=columns 在我的情况下不起作用。

import pymssql 
import pandas as pd 
 
 
con = pymssql.connect( 
     server="MSSQLSERVER", 
     port="1433", 
     user="us", 
     password="pass", 
     database="loc") 
 
cur = conn.cursor() 
 
cur.execute("SELECT id from ide") 
 
data=cur.fetchall() 
 
pandas=pd.DataFrame(data) 
 
cur.close() 
con.close() 

所以当我打印“pandas”时,结果是没有标题。那么我怎样才能收到它们呢?

请您参考如下方法:

我还发现向游标对象添加 as_dict=True 是可行的:

cur = conn.cursor(as_dict=True) 

然后你就可以运行剩下的了:

cur.execute("SELECT id from ide") 
data=cur.fetchall() 
pandas=pd.DataFrame(data) 

pandas 将从字典中获取列名