mysql批量导入函数
时间: 2023-11-23 17:56:22 浏览: 107
以下是一个使用Python编写的MySQL批量导入函数的示例:
```python
import csv
import MySQLdb
def load_csv_to_mysql(filename, tablename):
# 打开CSV文件并读取数据
with open(filename, 'r') as csvfile:
csvreader = csv.reader(csvfile)
headers = next(csvreader) # 获取表头
rows = []
for row in csvreader:
rows.append(row)
# 连接MySQL数据库
conn = MySQLdb.connect(host='localhost', user='root', passwd='password', db='database')
cursor = conn.cursor()
# 构造SQL语句并执行批量导入
sql = "INSERT INTO %s (%s) VALUES (%s)" % (tablename, ','.join(headers), ','.join(['%s']*len(headers)))
cursor.executemany(sql, rows)
# 提交更改并关闭连接
conn.commit()
cursor.close()
conn.close()
# 示例用法
load_csv_to_mysql('data.csv', 'mytable')
```
该函数使用Python的csv模块读取CSV文件中的数据,并使用MySQLdb模块连接MySQL数据库。然后,它构造一个批量导入的SQL语句,并使用executemany()方法执行批量导入。最后,它提交更改并关闭连接。
阅读全文