溫馨提示×

Python數(shù)據(jù)庫連接如何實現(xiàn)數(shù)據(jù)的批量操作

小樊
83
2024-11-09 14:17:51
欄目: 編程語言

在Python中，可以使用多種庫來連接數(shù)據(jù)庫并執(zhí)行批量操作。以下是一些常用的庫和方法：

  1. SQLite3：Python內(nèi)置的SQLite3庫適用于小型數(shù)據(jù)庫。使用executemany()方法可以執(zhí)行批量插入操作。
import sqlite3

# 連接數(shù)據(jù)庫
conn = sqlite3.connect('example.db')
cursor = conn.cursor()

# 創(chuàng)建表
cursor.execute('''CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY, name TEXT, age INTEGER)''')

# 批量插入數(shù)據(jù)
data = [
    ('Alice', 30),
    ('Bob', 25),
    ('Charlie', 22)
]
cursor.executemany('INSERT INTO users (name, age) VALUES (?, ?)', data)

# 提交事務(wù)
conn.commit()

# 查詢數(shù)據(jù)
cursor.execute('SELECT * FROM users')
print(cursor.fetchall())

# 關(guān)閉連接
conn.close()
  1. MySQL Connector/Python：這是一個流行的MySQL數(shù)據(jù)庫連接器，可以使用executemany()方法執(zhí)行批量插入操作。
import mysql.connector

# 連接數(shù)據(jù)庫
conn = mysql.connector.connect(
    host='localhost',
    user='your_username',
    password='your_password',
    database='example_db'
)
cursor = conn.cursor()

# 創(chuàng)建表
cursor.execute('''CREATE TABLE IF NOT EXISTS users (id INT AUTO_INCREMENT PRIMARY KEY, name VARCHAR(255), age INT)''')

# 批量插入數(shù)據(jù)
data = [
    ('Alice', 30),
    ('Bob', 25),
    ('Charlie', 22)
]
cursor.executemany('INSERT INTO users (name, age) VALUES (%s, %s)', data)

# 提交事務(wù)
conn.commit()

# 查詢數(shù)據(jù)
cursor.execute('SELECT * FROM users')
print(cursor.fetchall())

# 關(guān)閉連接
cursor.close()
conn.close()
  1. SQLAlchemy：這是一個功能強大的ORM庫，可以簡化數(shù)據(jù)庫操作。使用executemany()方法執(zhí)行批量插入操作。
from sqlalchemy import create_engine, Table, Column, Integer, String
from sqlalchemy.orm import sessionmaker

# 創(chuàng)建數(shù)據(jù)庫引擎
engine = create_engine('sqlite:///example.db')

# 定義表結(jié)構(gòu)
metadata = MetaData()
users = Table('users', metadata,
    Column('id', Integer, primary_key=True),
    Column('name', String),
    Column('age', Integer)
)

# 創(chuàng)建表
metadata.create_all(engine)

# 創(chuàng)建會話
Session = sessionmaker(bind=engine)
session = Session()

# 批量插入數(shù)據(jù)
data = [
    ('Alice', 30),
    ('Bob', 25),
    ('Charlie', 22)
]
session.execute(users.insert().values(data))

# 提交事務(wù)
session.commit()

# 查詢數(shù)據(jù)
result = session.query(users).all()
print(result)

# 關(guān)閉連接
session.close()

這些示例展示了如何使用不同的庫在Python中執(zhí)行批量數(shù)據(jù)庫操作。根據(jù)你的需求和使用的數(shù)據(jù)庫類型，可以選擇合適的庫和方法。

0 踩