欢迎访问宙启技术站
智能推送

使用Python进行数据库连接和操作

发布时间:2023-12-04 16:26:13

在Python中,可以使用多种方式连接和操作数据库。下面将介绍两种常用的方法:使用SQLite数据库和使用MySQL数据库。

1. 使用SQLite数据库

SQLite是一种嵌入式数据库引擎,不需要单独的数据库服务器。可以使用Python内置的sqlite3模块连接和操作SQLite数据库。

首先需要导入sqlite3模块:

import sqlite3

接下来可以使用connect()函数连接到SQLite数据库。如果数据库不存在,则会自动创建一个新的数据库文件:

conn = sqlite3.connect('example.db')

可以使用cursor()方法创建一个游标对象,该对象用于执行SQL语句:

cursor = conn.cursor()

执行SQL语句可以使用execute()方法,例如创建一个新的表:

cursor.execute('''CREATE TABLE stocks (date text, trans text, symbol text, qty real, price real)''')

插入数据可以使用execute()方法和commit()方法:

cursor.execute("INSERT INTO stocks VALUES ('2021-01-01', 'BUY', 'AAPL', 100, 138.23)")
conn.commit()

查询数据使用execute()方法和fetchone()方法或fetchall()方法:

cursor.execute("SELECT * FROM stocks")
print(cursor.fetchone())
print(cursor.fetchall())

最后,需要关闭数据库连接:

conn.close()

2. 使用MySQL数据库

MySQL是一种开源的关系型数据库管理系统,可以使用Python的第三方库mysql-connector-python连接和操作MySQL数据库。

首先需要安装mysql-connector-python库:

pip install mysql-connector-python

在代码中导入mysql.connector模块:

import mysql.connector

连接到MySQL数据库可以使用connect()函数:

conn = mysql.connector.connect(host='localhost', user='root', password='password', database='test')

可以使用cursor()方法创建一个游标对象,该对象用于执行SQL语句:

cursor = conn.cursor()

执行SQL语句可以使用execute()方法,例如创建一个新的表:

cursor.execute('''CREATE TABLE stocks (date DATE, trans VARCHAR(10), symbol VARCHAR(10), qty FLOAT, price FLOAT)''')

插入数据可以使用execute()方法和commit()方法:

cursor.execute("INSERT INTO stocks VALUES ('2021-01-01', 'BUY', 'AAPL', 100, 138.23)")
conn.commit()

查询数据使用execute()方法和fetchone()方法或fetchall()方法:

cursor.execute("SELECT * FROM stocks")
print(cursor.fetchone())
print(cursor.fetchall())

最后,需要关闭数据库连接:

conn.close()

以上是Python连接和操作SQLite和MySQL数据库的简单示例。实际使用中,还可以根据具体需要进行更复杂的数据库操作,如更新数据、删除数据等。