如何使用Python进行数据库操作_开发测试

创始人

2025-01-09 12:34:04

0次

在Python中进行数据库操作通常涉及以下几个步骤：安装数据库适配器（也称为数据库驱动或接口）、连接到数据库、执行SQL语句、处理结果以及关闭连接。下面我将以几个流行的数据库为例，介绍如何使用Python进行数据库操作。

SQLite是一种轻量级的数据库，它不需要独立的服务器进程或操作，可以直接在Python中使用。Python标准库中的sqlite3模块提供了对SQLite数据库的支持。

python复制代码

	`import sqlite3`

	`# 连接到SQLite数据库`
	`# 如果文件不存在，会自动在当前目录创建`
	`conn = sqlite3.connect('example.db')`

	`# 创建一个Cursor对象，并通过它执行SQL命令`
	`c = conn.cursor()`

	`# 创建表`
	`c.execute('''CREATE TABLE IF NOT EXISTS stocks`
	`(date text, trans text, symbol text, qty real, price real)''')`

	`# 插入数据`
	`c.execute("INSERT INTO stocks VALUES ('2023-01-01','BUY','RHAT',100,35.14)")`

	`# 提交事务`
	`conn.commit()`

	`# 查询数据`
	`c.execute('SELECT * FROM stocks WHERE symbol=?', ('RHAT',))`
	`print(c.fetchall())`

	`# 关闭连接`
	`conn.close()`

对于MySQL数据库，Python提供了mysql-connector-python和PyMySQL等第三方库来支持。这里以mysql-connector-python为例。

首先，你需要安装mysql-connector-python：

bash复制代码

pip install mysql-connector-python

然后，你可以像这样连接到MySQL数据库并执行操作：

python复制代码

	`import mysql.connector`

	`# 连接到MySQL数据库`
	`conn = mysql.connector.connect(`
	`host="localhost",`
	`user="yourusername",`
	`password="yourpassword",`
	`database="mydatabase"`
	`)`

	`c = conn.cursor()`

	`# 创建表`
	`c.execute("CREATE TABLE IF NOT EXISTS employees (id INT AUTO_INCREMENT PRIMARY KEY, name VARCHAR(255), position VARCHAR(255))")`

	`# 插入数据`
	`c.execute("INSERT INTO employees (name, position) VALUES (%s, %s)", ("John Doe", "Developer"))`

	`# 提交事务`
	`conn.commit()`

	`# 查询数据`
	`c.execute("SELECT * FROM employees")`
	`print(c.fetchall())`

	`# 关闭连接`
	`conn.close()`

对于PostgreSQL数据库，可以使用psycopg2库。

首先，安装psycopg2：

bash复制代码

pip install psycopg2-binary

然后，连接到PostgreSQL数据库并执行操作：

python复制代码

	`import psycopg2`

	`# 连接到PostgreSQL数据库`
	`conn = psycopg2.connect(`
	`dbname="mydatabase",`
	`user="myuser",`
	`password="mypassword",`
	`host="127.0.0.1",`
	`port="5432"`
	`)`

	`cur = conn.cursor()`

	`# 创建表`
	`cur.execute("CREATE TABLE IF NOT EXISTS test (id SERIAL PRIMARY KEY, num INTEGER, data VARCHAR(50))")`

	`# 插入数据`
	`cur.execute("INSERT INTO test (num, data) VALUES (%s, %s)", (100, "abc'def"))`

	`# 提交事务`
	`conn.commit()`

	`# 查询数据`
	`cur.execute("SELECT * FROM test")`
	`rows = cur.fetchall()`
	`for row in rows:`
	`print(row)`

	`# 关闭连接`
	`cur.close()`
	`conn.close()`

以上是在Python中操作SQLite、MySQL和PostgreSQL数据库的基本步骤。不同的数据库系统可能需要安装不同的库，并且连接字符串（用于连接到数据库的参数）也可能会有所不同。但是，一旦你建立了连接，执行SQL语句和处理结果的基本模式通常是相似的。