Python连接到PostgreSQL并查询数据的方法

Python环境

本文中使用的环境是Python 3.6.4。还需要提前安装psycopg2模块，此模块用于连接PostgreSQL数据库。

 

 

 

pip install psycopg2

 

连接到PostgreSQL数据库并查询数据

首先引用psycopg2模块：

import psycopg2

 

然后就可以通过如下方式连接到PostgreSQL数据库并执行查询数据的SQL语句：

# database，user，password，host，port分别对应要连接的PostgreSQL数据库的数据库名、数据库用户名、用户密码、主机、端口信息，请根据具体情况自行修改
conn = psycopg2.connect(database="test",user="postgres",password="postgres",host="127.0.0.1",port="5432")
cur = conn.cursor()
# 执行查询命令
cur.execute("SELECT name,setting FROM pg_settings")

 

获取查询结果

接着需要获取查询的结果，可以选择直接打印出来或者存放到文件中。

1）直接打印结果集

# 获取结果集的每一行
rows = cur.fetchall()
# 获取所有字段名
all_fields = cur.description


# 首先打印字段名
field_messages = []
for i in range(len(all_fields)):    
　　# 格式化输出结果，len参数是各列的显示宽度，可以指定常量，也可自定义函数进行获取。    
　　field_messages.append("{str:<{len}}".format(str=str(all_fields[i][0]), len=50))


field_message = "".join(field_messages)
print(field_message)


# 然后逐行打印结果集
for row in rows:    
　　row_messages = []    
　　for j in range(len(row)):        
　　　　# 格式化结果集        
　　　　row_messages.append("{str:<{len}}".format(str=str(row[j]), len=50))    
　　row_message = "".join(row_messages)    
　　print(row_message)

 

 

2）将结果集存放到文件中（以txt文件为例）

# 获取结果集的每一行
rows = cur.fetchall()
# 获取所有字段名
all_fields = cur.description
# 写入文件results.txt中，此处需要填写实际路径名
with open("results.txt", "w", encoding="utf-8") as f:    
　　# 首先写入字段名    
　　field_messages = []    
　　for i in range(len(all_fields)):        
　　　　# 格式化输出结果，len参数是各列的显示宽度，可以指定常量，也可自定义函数进行获取。        
　　　　field_messages.append("{str:<{len}}".format(str=str(all_fields[i][0]), len=50))    
　　field_message = "".join(field_messages)    
　　f.write(field_message + "\n")

　　# 然后写入结果集    
　　for row in rows:        
　　　　row_messages = []        
　　　　for j in range(len(row)):            
　　　　　　# 格式化结果集            
　　　　　　row_messages.append("{str:<{len}}".format(str=str(row[j]), len=50))        
　　　　row_message = "".join(row_messages)        
　　　　f.write(row_message + "\n")

 

关闭数据库连接

所有操作完成之后需要关闭对PostgreSQL数据库的连接。

# 关闭连接
conn.close()