在Python中,可以使用數據庫連接庫(如pymysql
、psycopg2
、sqlite3
等)來連接數據庫,并將數據批量寫入數據庫。以下是一個示例代碼,使用pymysql
庫將數據批量寫入MySQL數據庫:
import pymysql
# 連接數據庫
connection = pymysql.connect(
host='localhost',
user='username',
password='password',
database='database_name'
)
# 獲取游標
cursor = connection.cursor()
# 批量插入數據
data = [
('John', 25, 'john@example.com'),
('Alice', 30, 'alice@example.com'),
('Bob', 35, 'bob@example.com')
]
sql = "INSERT INTO users (name, age, email) VALUES (%s, %s, %s)"
cursor.executemany(sql, data)
# 提交事務
connection.commit()
# 關閉游標和連接
cursor.close()
connection.close()
在上述示例中,首先使用pymysql
庫連接數據庫,然后獲取游標。接下來,定義要插入的數據列表data
,其中每個元素表示一條記錄。然后,使用executemany
方法執行批量插入操作,將數據列表和SQL語句傳遞給executemany
方法。最后,使用commit
方法提交事務,并關閉游標和連接。
需要根據實際情況修改數據庫連接的相關參數(如host
、user
、password
、database
等),以及表名、字段名和要插入的數據。另外,如果使用的是其他數據庫連接庫,可以根據庫的文檔進行相應的調整。