首页 > 编程语言 > 详细

Python:使用Faker随机生成测试数据

时间:2020-07-09 15:50:27      阅读:86      评论:0      收藏:0      [点我收藏+]

工作中有时需要一些测试的假数据,比如姓名、手机号、身份证号等,使用Faker就可以随机生成了,可以保存到本地excel文件,也可以存入sql数据库。

比如先构造5条用户的基本信息,包含字段:姓名、手机号、身份证号,保存到本地excel文件。试一下~~

(需要提前下载安装faker、xlwt、pandas等)

from faker import Faker
import xlwt
import pandas as pd

fake = Faker(‘zh_CN‘)


def save_to_excel():
    work_book = xlwt.Workbook(encoding=‘utf-8‘)
    work_sheet = work_book.add_sheet(‘测试数据01‘)

    # 添加字段名
    head = [‘姓名‘, ‘手机号‘, ‘身份证号‘]
    for h in range(len(head)):
        work_sheet.write(0, h, head[h])

    # 添加构造的随机数据
    for i in range(5):
        name = fake.name()
        phone = fake.phone_number()
        id_card = fake.ssn()
        work_sheet.write(i+1, 0, name)
        work_sheet.write(i+1, 1, phone)
        work_sheet.write(i+1, 2, id_card)

    work_book.save(‘测试数据01.xls‘)


def save_to_excel2(file_path, n):
    res = []
    for i in range(n):
        res.append([fake.name(), fake.phone_number(), fake.ssn()])

    # list转dataFrame
    df = pd.DataFrame(data=res, columns=[‘姓名‘, ‘手机号‘, ‘身份证号‘])

    # 保存到本地excel
    df.to_excel(file_path, index=False)

if __name__ == ‘__main__‘:
    save_to_excel()

运行成功后在项目所在目录下生成一个excel文件,文件名就是:测试数据01.xls

技术分享图片

打开excel文件的效果是这样的(当然数据都是假的啦,仅用于测试):

技术分享图片

Python:使用Faker随机生成测试数据

原文:https://www.cnblogs.com/kechen/p/13274003.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!