sql读取数据库，查询的数据转为dataframe

最新推荐文章于 2024-05-27 18:45:58 发布

原创最新推荐文章于 2024-05-27 18:45:58 发布 · 3.2k 阅读

10 ·

CC 4.0 BY-SA版权

本文介绍如何使用Python执行SQL查询并获取全部数据，包括利用游标的description属性获取列名，将查询结果转换为Pandas DataFrame，并保存为CSV文件。

部署运行你感兴趣的模型镜像

# 执行sql语句
cs1.execute('select * from tabel_name')

# 拿到所有数据
data = cs1.fetchall()
print(data)

# 拿到所有数据的描述
columnDes = cs1.description
print(columnDes)

# 通过描述，得到列名称
columnNames = [columnDes[i][0] for i in range(len(columnDes))]
print(columnNames)

# 得到的数据是元组，元组要转为list，list再次转为dataframe
df = pd.DataFrame(list(data))
# 上一步得到的df的列名称为默认的0、1、2、34、5、6、7、8、9....
# 修改df列名称为数据库里的列名称
df.columns = columnNames
print(df)

df.to_csv("../file/test_sql_1csv", index=False)

您可能感兴趣的与本文相关的镜像