python连接mysql循环插入千万条数据脚本

时间：2022-03-31 17:53:14

之前都是在mysql的存储过程中插入数据，毕竟mysql语法函数有限，很多都有限制。突然想到学了python正好可以练练手。
首先需要安装pymysql模块包（模块包安装请自行百度）

pip install pymysql

mysql语句的符号一定要看清楚，否则运行不生成数据，运行结果又没有错误提示很难定位问题，博主就因为一个符号浪费大量的时间。
不多说了，直接上代码（注意符合，注意符合，注意符合！！！）

#encoding:utf-8

import pymysql

import time

import random

#连接mysql

db = pymysql.connect(host='14.29.249.74',port=3306,user='kx_admin',passwd='密码',db='ilock')

#获取mysql操作光标

cursor = db.cursor()

#初始化变量

count = 0

#设置sql语句循环次数

while count <= 100:

    count += 1

    #定义mysql字段的范围随机数变量

    num = random.randint(0,2)

    memo_num = random.randint(100,111)

    city_list = ['长沙','湘潭','株洲','衡阳']

    a = random.choice(city_list)

    #生成mysql语句插入语句

    sql ="insert  into iodn_city(city_code,city_name,memo)values({},'{}',{})".format(num,a,memo_num)#执行sql语句

    try:

        cursor.execute(sql)

        db.commit()

    #错误回滚

    except:

        db.rollback()

#关闭mysql

db.close()

相关文章