import pymysql
from faker import Faker
import random
conn = pymysql.connect(
host='主机名或IP地址',
user='用户名',
password='密码',
database='数据库名',
port=3306 # 端口
)
# 创建游标对象
cur = conn.cursor()
# 创建faker对象,用于生成数据
faker = Faker('zh_CN')
# 定义要插入的表名
table_name = 'table_name'
def generate_and_insert_data(table_name,conn,cur,num_rows):
job_id1 = ["job001","job002","job003"]
# 生成列名
columns = '(user_id,user_name,user_age,job_id,job_age,job_enable,city,work_date)'
# 生成值
values = [(faker.pystr(min_chars=8,max_chars=8).upper()+faker.numerify(),
faker.name(),
random.randint(23,55),
random.choice(job_id1),
faker.random_int(min=1,max=10),
random.choice(['true','false']),
faker.city(),
faker.date_between(start_date='-10y',end_date='-1m')) for i in range(1,num_rows+1)]
# 生成批量插入的SQL语句
insert_query = 'INSERT INTO {} {} VALUES %s'.format(table_name,columns)
# 创建批量插入的数据元组列表
execute_values = [(v,) for v in values]
# 执行批量插入
cur.executemany(insert_query,execute_values)
# 提交事务
conn.commit()
# 调用函数,生成并插入数据
num = 10
generate_and_insert_data(table_name,conn,cur,num)
# 关闭游标和连接
cur.close()
conn.close()
print(f'{num}条数据已插入数据库中...')
python faker批量生成数据写入MySQL
于 2024-06-15 17:18:33 首次发布