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

最新推荐文章于 2024-03-30 10:00:00 发布

他强任他强03

最新推荐文章于 2024-03-30 10:00:00 发布

阅读量374

点赞数

分类专栏： python 文章标签： python postgresql sql

本文链接：https://blog.csdn.net/nansen0324/article/details/117039785

版权

python 专栏收录该内容

9 篇文章 0 订阅

订阅专栏

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()