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

Python编写的简单数据库操作案例

发布时间:2023-12-04 21:24:42

数据库是存储、管理和检索数据的工具,可以将数据存储在计算机的硬盘上,并提供一套机制来操作这些数据。Python提供了多个数据库操作的模块,如MySQLdb、sqlite3等,可以方便地对数据库进行操作。

下面以MySQL为例,介绍Python编写的简单数据库操作案例。

首先,我们需要安装MySQL的Python驱动,可以使用pip命令进行安装:

pip install mysql-connector-python

然后,我们需要连接到数据库。首先,需要导入mysql.connector模块,并使用connect函数连接到数据库。在连接数据库时,我们需要提供数据库的地址、用户名、密码等信息。

import mysql.connector

# 连接数据库
cnx = mysql.connector.connect(user='root', password='password', host='127.0.0.1', database='test')

数据库连接成功后,我们可以使用cursor方法创建一个游标对象,该对象可以执行SQL语句。

# 创建游标对象
cursor = cnx.cursor()

接下来,我们可以创建一个数据表。在创建数据表之前,我们需要定义数据表的结构,包括数据表的名称、字段的名称和字段的数据类型等信息。下面是一个创建学生表的示例:

# 创建学生表
create_table = """
CREATE TABLE IF NOT EXISTS students (
    id INT AUTO_INCREMENT PRIMARY KEY,
    name VARCHAR(50) NOT NULL,
    age INT NOT NULL,
    gender ENUM('M', 'F') NOT NULL,
    department VARCHAR(100) NOT NULL
)
"""

cursor.execute(create_table)

然后,我们可以向数据表插入数据。在插入数据之前,我们需要定义要插入的数据。下面是一个插入学生数据的示例:

# 插入学生数据
insert_data = """
INSERT INTO students (name, age, gender, department)
VALUES (%s, %s, %s, %s)
"""

data = [
    ('Tom', 20, 'M', 'Computer Science'),
    ('Jane', 21, 'F', 'Mathematics'),
    ('John', 19, 'M', 'Physics')
]

cursor.executemany(insert_data, data)

# 提交事务
cnx.commit()

我们还可以使用SELECT语句来查询数据。下面是一个查询学生数据的示例:

# 查询学生数据
select_data = "SELECT * FROM students"

cursor.execute(select_data)

for (id, name, age, gender, department) in cursor:
    print(f"ID: {id}, Name: {name}, Age: {age}, Gender: {gender}, Department: {department}")

最后,我们需要关闭游标和数据库连接:

# 关闭游标
cursor.close()

# 关闭数据库连接
cnx.close()

以上就是一个简单的使用Python进行数据库操作的案例。我们可以根据实际需求,灵活地使用数据库操作模块进行其他操作,如更新数据、删除数据等等。