python如何插入数据库

Python插入数据库的方式有多种，包括使用SQL语句、ORM框架以及不同的数据库连接库。常见的方法包括使用SQLite、MySQL、PostgreSQL、SQLAlchemy等。推荐使用SQLAlchemy进行数据库操作，因为它提供了更高层次的抽象和更灵活的操作。

SQLAlchemy提供了ORM（对象关系映射）功能，使得开发者可以用Python对象来操作数据库，避免直接编写SQL语句。

接下来，我们将详细介绍Python如何插入数据库，包括使用SQLite、MySQL、PostgreSQL和SQLAlchemy的方式。

一、使用SQLite插入数据库

SQLite是一个轻量级的嵌入式数据库，适合小型应用和快速开发。Python标准库自带sqlite3模块，可直接使用。

1.1 创建连接和表

首先，我们需要创建一个数据库连接，并创建一个示例表。

import sqlite3
连接到SQLite数据库
connection = sqlite3.connect('example.db')
cursor = connection.cursor()
创建表
cursor.execute('''
    CREATE TABLE IF NOT EXISTS users (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        name TEXT,
        age INTEGER
    )
''')
提交事务
connection.commit()

1.2 插入数据

接下来，我们将数据插入到表中。

# 插入数据
cursor.execute('''
    INSERT INTO users (name, age) VALUES (?, ?)
''', ('Alice', 30))
提交事务
connection.commit()

1.3 批量插入数据

如果需要批量插入数据，可以使用executemany方法。

# 批量插入数据
data = [('Bob', 25), ('Charlie', 35), ('David', 40)]
cursor.executemany('''
    INSERT INTO users (name, age) VALUES (?, ?)
''', data)
提交事务
connection.commit()

1.4 关闭连接

最后，关闭数据库连接。

# 关闭连接
cursor.close()
connection.close()

二、使用MySQL插入数据库

MySQL是一个流行的关系型数据库管理系统，适合中大型应用。Python连接MySQL通常使用mysql-connector-python库。

2.1 安装mysql-connector-python

首先，安装mysql-connector-python库。

pip install mysql-connector-python

2.2 创建连接和表

然后，创建一个数据库连接，并创建一个示例表。

import mysql.connector
连接到MySQL数据库
connection = mysql.connector.connect(
    host='localhost',
    user='yourusername',
    password='yourpassword',
    database='testdb'
)
cursor = connection.cursor()
创建表
cursor.execute('''
    CREATE TABLE IF NOT EXISTS users (
        id INT AUTO_INCREMENT PRIMARY KEY,
        name VARCHAR(255),
        age INT
    )
''')
提交事务
connection.commit()

2.3 插入数据

接下来，我们将数据插入到表中。

# 插入数据
cursor.execute('''
    INSERT INTO users (name, age) VALUES (%s, %s)
''', ('Alice', 30))
提交事务
connection.commit()

2.4 批量插入数据

如果需要批量插入数据，可以使用executemany方法。

# 批量插入数据
data = [('Bob', 25), ('Charlie', 35), ('David', 40)]
cursor.executemany('''
    INSERT INTO users (name, age) VALUES (%s, %s)
''', data)
提交事务
connection.commit()

2.5 关闭连接

最后，关闭数据库连接。

# 关闭连接
cursor.close()
connection.close()

三、使用PostgreSQL插入数据库

PostgreSQL是一个功能强大、开源的对象关系型数据库系统，适合复杂和高性能应用。Python连接PostgreSQL通常使用psycopg2库。

3.1 安装psycopg2

首先，安装psycopg2库。

pip install psycopg2

3.2 创建连接和表

然后，创建一个数据库连接，并创建一个示例表。

import psycopg2
连接到PostgreSQL数据库
connection = psycopg2.connect(
    host='localhost',
    database='testdb',
    user='yourusername',
    password='yourpassword'
)
cursor = connection.cursor()
创建表
cursor.execute('''
    CREATE TABLE IF NOT EXISTS users (
        id SERIAL PRIMARY KEY,
        name VARCHAR(255),
        age INT
    )
''')
提交事务
connection.commit()

3.3 插入数据

接下来，我们将数据插入到表中。

# 插入数据
cursor.execute('''
    INSERT INTO users (name, age) VALUES (%s, %s)
''', ('Alice', 30))
提交事务
connection.commit()

3.4 批量插入数据

如果需要批量插入数据，可以使用executemany方法。

# 批量插入数据
data = [('Bob', 25), ('Charlie', 35), ('David', 40)]
cursor.executemany('''
    INSERT INTO users (name, age) VALUES (%s, %s)
''', data)
提交事务
connection.commit()

3.5 关闭连接

最后，关闭数据库连接。

# 关闭连接
cursor.close()
connection.close()

四、使用SQLAlchemy插入数据库

SQLAlchemy是一个Python SQL工具包和对象关系映射（ORM）库，提供了对数据库操作的更高级抽象，支持多种数据库。

4.1 安装SQLAlchemy

首先，安装SQLAlchemy库。

pip install sqlalchemy

4.2 创建连接和表

然后，创建一个数据库连接，并定义一个映射类来表示表。

from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
创建连接
engine = create_engine('sqlite:///example.db', echo=True)
Base = declarative_base()
定义映射类
class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True, autoincrement=True)
    name = Column(String)
    age = Column(Integer)
创建表
Base.metadata.create_all(engine)

4.3 插入数据

接下来，我们将数据插入到表中。

# 创建会话
Session = sessionmaker(bind=engine)
session = Session()
插入数据
new_user = User(name='Alice', age=30)
session.add(new_user)
提交事务
session.commit()

4.4 批量插入数据

如果需要批量插入数据，可以使用add_all方法。

# 批量插入数据
new_users = [
    User(name='Bob', age=25),
    User(name='Charlie', age=35),
    User(name='David', age=40)
]
session.add_all(new_users)
提交事务
session.commit()