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

工作中有时需要一些测试的假数据，比如姓名、手机号、身份证号等，使用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文件的效果是这样的（当然数据都是假的啦，仅用于测试）：