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

Python数据库函数:如何连接数据库和进行操作

发布时间:2023-09-24 14:46:02

Python提供了多种库和模块来连接和操作数据库。其中最常用的有sqlite3MySQLdbpsycopg2pymssql等。本文以sqlite3为例,介绍如何连接数据库和进行操作。

首先,需要安装对应的库。使用pip命令来安装sqlite3库:

pip install sqlite3

安装完库之后,可以使用以下代码来连接数据库:

import sqlite3

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

sqlite3.connect()函数用于连接数据库,参数是数据库的路径。如果该路径下的数据库不存在,则会自动创建一个新的数据库。

连接数据库成功之后,可以使用以下代码来创建表和插入数据:

# 创建表
conn.execute('''CREATE TABLE IF NOT EXISTS Person
                (ID INT PRIMARY KEY     NOT NULL,
                NAME           TEXT    NOT NULL,
                AGE            INT     NOT NULL);''')

# 插入数据
conn.execute("INSERT INTO Person (ID,NAME,AGE) \
                VALUES (1, 'John Doe', 25)")

# 提交事务
conn.commit()

上述代码中,首先使用conn.execute()来执行SQL语句,用于创建表和插入数据。创建表的SQL语句使用了CREATE TABLE语法,用于创建名为Person的表,表中包含ID、NAME、AGE三个列。插入数据的SQL语句使用了INSERT INTO语法,将数据插入到Person表中。

最后,使用conn.commit()函数来提交事务。在进行数据库操作之后,必须使用该函数将改变保存到数据库中。

除了插入数据之外,还可以使用以下代码来查询数据:

# 查询数据
cursor = conn.execute("SELECT * FROM Person")
for row in cursor:
    print("ID = ", row[0])
    print("NAME = ", row[1])
    print("AGE = ", row[2])

上述代码中,使用conn.execute()函数执行SELECT语句来查询数据,并将结果保存在cursor对象中。接着使用for循环遍历cursor对象来输出查询结果。

最后,使用以下代码来关闭数据库连接:

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

上述代码中,使用conn.close()函数来关闭数据库连接,释放资源。

综上所述,通过以上代码,可以连接数据库并进行操作。除了上述的示例代码之外,还可以进行更新数据、删除数据等操作,具体使用方法可以参考相关库的文档。