python faker批量生成数据写入MySQL

k_means123

已于 2024-06-15 17:27:31 修改

阅读量133

点赞数 1

分类专栏： python 文章标签： python mysql

于 2024-06-15 17:18:33 首次发布

本文链接：https://blog.csdn.net/k_means123/article/details/139705349

版权

python 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

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}条数据已插入数据库中...')