MySQL Python+MySQL – Bulk Insert

在开发中，有时用户需要一次向数据库中插入多条记录，避免了反复连接数据库和操作数据库的开销。这时就需要使用MySQL的批量插入方式。

阅读更多：MySQL 教程

批量插入方式

在MySQL中，有两种批量插入方式：

使用多条INSERT语句，一条语句对应一条记录；
使用INSERT INTO … VALUES (value1, value2), (value1, value2), (value1, value2)…语句，一条语句对应多条记录。

使用前者需要反复连接和关闭数据库，效率低；使用后者，只需要连接一次数据库，可以一次插入多条数据，效率高。

Python使用MySQL批量插入

Python中，批量插入需要用到pymysql库，该库是基于Python3，可以通过pip install pymysql安装。

示例代码如下：

import pymysql


def bulk_insert():
    conn = pymysql.connect(
        host='localhost',
        port=3306,
        user='root',
        password='root',
        database='test_db',
        charset='utf8'
    )
    cursor = conn.cursor()

    values = [
        ('Tom', 20),
        ('Jerry', 21),
        ('Ricky', 22)
    ]

    try:
        sql = 'INSERT INTO student (name, age) VALUES %s'
        cursor.executemany(sql, values)
        conn.commit()
        print('Success')
    except Exception as e:
        conn.rollback()
        print('Failed', str(e))
    finally:
        cursor.close()
        conn.close()


if __name__ == '__main__':
    bulk_insert()