Python-实现对表插入百万条数据

时间:2024-10-18 14:33:14

新手小白写的,我自己都不知道对不对,先写下来记着,以后掌握更多Python知识后,再来看下写的对不对。

题目:造一百万条大学生的基本信息(学校、专业、姓名、学号)

分析思路:利用Python的MySQL模块实现,首先导入MySQL模块,创建一个student表,再利用循环插入100万条学生记录。(我知道这里应该有优化,可是我还不知道怎么实现,哭------)

代码实现:

import MySQLdb

db=MySQLdb.connect('localhost','user','123','testdb')

cursor=db.cursor()

sql1='''
          create table student(
          school char(20),
          major char(10),
          sname char(10);
          snum int primary key)'''
cursor.execute(sql1)                #执行建表操作

for i in range(1000000)            #循环执行1000000次
    try:

school=input('请输入学校名称:')

major=input('请输入专业名称:')

sname=input('请输入学生姓名:')

snum=input('请输入学生学号:')

sql2=insert into student(school,major,sname,snum)

         cursor.execute(sql2)

         db.commit

    except:

          db.rollback

db.close()