工作中性能测试,或者自动化测试有时候需要造大量的数据,下面用python编写一个脚本可以直接批量生成数据并插入excel当中。
主要用到了python的Faker库和Pandas库
Faker:创建各种各样的伪数据
Pandas:是基于Numpy的专业数据分析工具, 可以灵活高效的处理各种数据集,它提供了的数据结构有DataFrame和Series等,我们可以简单粗的把DataFrame理解为Excel里面的一张表,而Series就是表中的某一列
1.先安装Faker和Pandas
pip install Faker
pip install pandas
或者在PyCharm中,file-settings进行安装Faker和Pandas 如下所示:
代码如下:
import random
from faker import Faker
import pandas as pd
fk=Faker(locale='zh_CN')
Faker.seed(0)
def getdata():
num_list=["姓名","年龄","地址","出生年月","手机号","邮箱"]
#获取姓名
name=fk.name()
#随机生成年龄
age=random.randint(1,100)
#获取地址
address=fk.address()
#先获取省份证号,然后截取出生年月
birthdate=fk.ssn()[6:14]
print(birthdate)
#获取电话号码
number=fk.phone_number()
#获取邮箱
email=fk.email()
info_list=[name,age,address,birthdate,number,email]
my_info=dict(zip(num_list,info_list))
# print(my_info)
return my_info
#定义列标题
df=pd.DataFrame(columns=["姓名","年龄","地址","出生年月","手机号","邮箱"])
for i in range(100):
my_info=[getdata()]
#(my_info)
df1=pd.DataFrame(my_info)
#print(df1)
#合并数据
df=pd.concat([df,df1])
#print(df)
#写入excel
df.to_excel("D:\\test\\lxj\\测试数据.xlsx",index=None)
生成的EXCEL: