python中查询数据库

原创

mob649e8158a948 2023-10-07 13:02:27 ©著作权

文章标签 数据库 Python sqlite 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob649e8158a948的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python中查询数据库

在使用Python进行开发时，经常需要与数据库进行交互，其中查询数据库是一项基础且重要的操作。Python提供了多个库来实现数据库查询的功能，如sqlite3、pymysql、psycopg2等。本文将介绍如何使用Python查询数据库的基本步骤和常用的库。

数据库查询的基本步骤

通常，数据库查询的基本步骤如下：

连接数据库：使用合适的库连接到目标数据库。
创建游标：创建一个游标对象，用于执行SQL查询。
执行查询：使用游标对象执行SQL查询。
获取查询结果：从游标对象中获取查询结果。
关闭游标和数据库连接：关闭游标和数据库连接，释放资源。

下面我们将以sqlite3库为例，演示如何使用Python查询SQLite数据库。

使用sqlite3查询数据库

sqlite3库是Python内置的轻量级数据库库，适用于SQLite数据库。下面是使用sqlite3库查询数据库的示例代码：

import sqlite3

# 连接数据库
conn = sqlite3.connect('example.db')

# 创建游标
cursor = conn.cursor()

# 执行查询
cursor.execute('SELECT * FROM employees')

# 获取查询结果
result = cursor.fetchall()

# 打印查询结果
for row in result:
    print(row)

# 关闭游标和数据库连接
cursor.close()
conn.close()

以上代码首先通过sqlite3.connect()方法连接到名为example.db的SQLite数据库，然后使用conn.cursor()方法创建游标对象。接下来，使用游标对象的execute()方法执行SQL查询，这里我们查询了employees表的所有记录。然后，使用fetchall()方法获取查询结果，并使用循环遍历打印每一行记录。最后，通过调用cursor.close()和conn.close()方法关闭游标和数据库连接，释放资源。

使用其他数据库库查询数据库

除了sqlite3库外，Python还提供了其他库用于查询各种类型的数据库。下面是使用pymysql库查询MySQL数据库和使用psycopg2库查询PostgreSQL数据库的示例代码：

使用pymysql库查询MySQL数据库的示例代码：

import pymysql

# 连接数据库
conn = pymysql.connect(host='localhost', user='root', password='password', database='example')

# 创建游标
cursor = conn.cursor()

# 执行查询
cursor.execute('SELECT * FROM employees')

# 获取查询结果
result = cursor.fetchall()

# 打印查询结果
for row in result:
    print(row)

# 关闭游标和数据库连接
cursor.close()
conn.close()

使用psycopg2库查询PostgreSQL数据库的示例代码：

import psycopg2

# 连接数据库
conn = psycopg2.connect(host='localhost', user='postgres', password='password', database='example')

# 创建游标
cursor = conn.cursor()

# 执行查询
cursor.execute('SELECT * FROM employees')

# 获取查询结果
result = cursor.fetchall()

# 打印查询结果
for row in result:
    print(row)

# 关闭游标和数据库连接
cursor.close()
conn.close()

以上代码中，分别使用pymysql库和psycopg2库连接到MySQL和PostgreSQL数据库，并执行相同的查询操作。其余的步骤和sqlite3库的使用方式基本相同。