python操作数据库(PostgreSQL)并将数据(包含列名)写到csv档，以“,”分隔

最新推荐文章于 2025-11-02 15:33:08 发布

原创最新推荐文章于 2025-11-02 15:33:08 发布 · 459 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#python #postgresql #sql

python 专栏收录该内容

9 篇文章

订阅专栏

本文介绍如何利用Python中的psycopg2模块连接PostgreSQL数据库，并执行SQL查询将结果保存为CSV文件。通过示例代码展示了连接数据库、执行查询及结果导出的具体步骤。

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

import psycopg2
import boto3

stmt = """
SELECT
    film.film_id,
    title,
    inventory_id
FROM
    film
LEFT JOIN inventory 
    ON inventory.film_id = film.film_id
ORDER BY title;
"""
#创建数据库连接
db_conn = psycopg2.connect(host='数据库连接ip', dbname='数据库schema', user='用户名', password='密码')
print('Connect [postgres] database successfully!')

# 创建连接数据库的游标
cur = db_conn.cursor()
# 执行sql语句
cur.execute(stmt)
# 串出的表头部分
header = [row[0] for row in cur.description]
# 串出的表的数据部分
rows = cur.fetchall()
print(header)
print("------------------")
print(rows)

# 创建task2.csv文件
f = open('task2' + '.csv', 'w')
# 将表头写进csv档并以","分隔
f.write(','.join(header) + '\n')
# 将数据写进csv并以","分隔
for row in rows:
    f.write(','.join(str(r) for r in row) + '\n')

f.close()
cur.close()