使用Python进行数据库连接和操作
在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数据库的简单示例。实际使用中,还可以根据具体需要进行更复杂的数据库操作,如更新数据、删除数据等。
