数据库连接和查询的Python函数
发布时间:2023-08-28 07:16:38
在Python中,我们可以使用多种库和模块来连接和查询数据库。下面我们将简要介绍一些常用的库和函数。
1. mysql.connector
mysql.connector是一个用于连接和操作MySQL数据库的Python库。可以使用以下代码来连接到MySQL数据库并执行查询:
import mysql.connector # 创建数据库连接 cnx = mysql.connector.connect(user='username', password='password', host='host', database='database') # 创建游标对象 cursor = cnx.cursor() # 执行查询 query = "SELECT * FROM table" cursor.execute(query) # 获取查询结果 result = cursor.fetchall() # 关闭游标和连接 cursor.close() cnx.close()
2. sqlite3
sqlite3是Python标准库中的一个模块,用于连接和操作SQLite数据库。可以使用以下代码来连接到SQLite数据库并执行查询:
import sqlite3
# 创建数据库连接
conn = sqlite3.connect('database.db')
# 创建游标对象
cursor = conn.cursor()
# 执行查询
query = "SELECT * FROM table"
cursor.execute(query)
# 获取查询结果
result = cursor.fetchall()
# 关闭游标和连接
cursor.close()
conn.close()
3. PostgreSQL
对于连接和查询PostgreSQL数据库,我们可以使用psycopg2库。可以使用以下代码来连接到PostgreSQL数据库并执行查询:
import psycopg2 # 创建数据库连接 conn = psycopg2.connect(user='username', password='password', host='host', port='port', database='database') # 创建游标对象 cursor = conn.cursor() # 执行查询 query = "SELECT * FROM table" cursor.execute(query) # 获取查询结果 result = cursor.fetchall() # 关闭游标和连接 cursor.close() conn.close()
上述代码中的query变量是要执行的SQL查询语句。执行cursor.execute(query)语句后,可以通过cursor.fetchall()获取查询结果。
以上是连接和查询数据库的Python函数的简要介绍。当连接不同类型的数据库时,需要使用相应的库和模块,并提供正确的连接参数。执行查询后,可以通过游标对象获取查询结果。
