Python随机生成数据后插入到PostgreSQL

204 阅读 0 评论 135 点赞

我是靠谱客的博主傲娇西装，这篇文章主要介绍Python随机生成数据后插入到PostgreSQL，现在分享给大家，希望可以做个参考。

用Python随机生成学生姓名，三科成绩和班级数据，再插入到PostgreSQL中。

模块用psycopg2 random

复制代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
import random
import psycopg2

fname=['金','赵','李','陈','许','龙','王','高','张','侯','艾','钱','孙','周','郑']
mname=['玉','明','玲','淑','偑','艳','大','小','风','雨','雪','天','水','奇','鲸','米','晓','泽','恩','葛','玄','道','振','隆','奇']
lname=['','玲','','芳','明','红','国','芬','','云','娴','隐','','花','叶','','黄','亮','锦','茑','军','','印','','凯']

# 数据库连接参数
conn = psycopg2.connect(database="girl", user="jm", password="123", host="127.0.0.1", port="5432")
cur = conn.cursor()

for x in range(200):
  #生成随机数据  
  pname=random.choice(fname)+random.choice(mname)+random.choice(lname)
  math=random.randint(40,100)
  english=random.randint(40,100)
  chinese=random.randint(40,100)
  pclass=random.randint(1,3)
  
  #插入数据 （特别注意只能用%s 不能用%d,数值型数据不用引号
  cur.execute("insert into score values(%s,%s,%s,%s,%s,%s)" ,(x,pname,math,english,chinese,glass))
  
  conn.commit()  #提交命令，否则数据库不执行插入操作
  
cur.close()
conn.close()