带你轻松掌握6种 Python 数据库操作及代码案例

在数据处理和管理领域，Python作为一种高效、易用的编程语言，拥有丰富的数据库操作模块，可以轻松实现对关系型数据库的数据操作。

本文将介绍六种常见的Python数据库操作模块，并提供相应的代码案例，帮助读者快速上手。

推荐

干货满满！史上最全 Python 数据分析学习路线来了

如果你还在初学阶段，推荐你按照上面流程。

一、MySQL数据库：pymysql

pymysql是Python操作MySQL数据库的重要模块，它提供了丰富的API和功能，可以实现数据库的连接、查询、插入、更新等操作。

以下是一个简单的代码示例：

import pymysql

# 连接数据库
conn = pymysql.connect(host='localhost', user='root', password='123456', database='test')

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

# 执行SQL语句
cursor.execute("SELECT * FROM students")

# 获取查询结果
result = cursor.fetchall()

# 打印结果
for row in result:
    print(row)

# 关闭游标和连接
cursor.close()
conn.close()

二、SQLite数据库：sqlite3

sqlite3是Python内置的轻量级数据库模块，适用于小型项目和嵌入式设备。

以下是一个简单的代码示例：

import sqlite3

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

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

# 执行SQL语句
cursor.execute("SELECT * FROM students")

# 获取查询结果
result = cursor.fetchall()

# 打印结果
for row in result:
    print(row)

# 关闭游标和连接
cursor.close()
conn.close()

三、PostgreSQL数据库：psycopg2

psycopg2是Python操作PostgreSQL数据库的模块，它提供了高性能和稳定的数据库连接和操作功能。

以下是一个简单的代码示例：

import psycopg2

# 连接数据库
conn = psycopg2.connect(database="test", user="postgres", password="123456", host="localhost", port="5432")

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

# 执行SQL语句
cursor.execute("SELECT * FROM students")

# 获取查询结果
result = cursor.fetchall()

# 打印结果
for row in result:
    print(row)

# 关闭游标和连接
cursor.close()
conn.close()

四、Oracle数据库：cx_Oracle

cx_Oracle是Python操作Oracle数据库的模块，它提供了完整的Oracle数据库连接和操作功能。

以下是一个简单的代码示例：

import cx_Oracle

# 连接数据库
conn = cx_Oracle.connect("username/password@localhost:1521/orcl")

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

# 执行SQL语句
cursor.execute("SELECT * FROM students")

# 获取查询结果
result = cursor.fetchall()

# 打印结果
for row in result:
    print(row)

# 关闭游标和连接
cursor.close()
conn.close()

五、MongoDB数据库：pymongo

pymongo是Python操作MongoDB数据库的模块，它提供了简单易用的API和功能，适用于处理非结构化数据。

以下是一个简单的代码示例：

from pymongo import MongoClient

# 连接数据库
client = MongoClient('mongodb://localhost:27017/')

# 获取数据库
db = client['test']

# 获取集合
collection = db['students']

# 查询数据
result = collection.find()

# 打印结果
for doc in result:
    print(doc)

# 关闭连接
client.close()

六、Redis数据库：redis

redis是一种高性能的键值对存储数据库，Python的redis模块提供了对redis数据库的连接和操作功能。

以下是一个简单的代码示例：

import redis

# 连接数据库
r = redis.Redis(host='localhost', port=6379, db=0)

# 设置键值对
r.set('name', 'Tom')

# 获取键值对
value = r.get('name')

# 打印结果
print(value)

# 关闭连接
r.close()

以上就是六种常见的Python操作数据库数据的模块及相应的代码案例。

读者可以根据自己的需求选择合适的模块进行数据库操作，提高数据处理和管理的效率。

希望本文对读者有所帮助！