各位大牛,我想生成按照一定的比例生成测试数据,比如我想生成1000条人员数据,其中的属性包括性别,学历,年龄等。其中男女比例为6:4,学历上初中、高中和大学的比例为3:3:4。
想生成这样的数据不知道我该怎么实现,恳请各位大佬指点一下,最好是用python实现,十分感激。
![](https://profile-avatar.csdnimg.cn/default.jpg!4)
python学习生成测试数据
- 写回答
- 好问题 0 提建议
- 追加酬金
- 关注问题
- 邀请回答
-
2条回答 默认 最新
- Edward_LA 2019-10-27 15:18关注
先上代码,在‘Specify Info’内可以定义你想要的比例信息,但是要确保你给出的各项比例值相加应该是相等的。
运行改代码后生成‘sample.csv’文件,里面包含编号,性别,年龄,学历。
Python 3.7# Import import pandas as pd import random # Specify Info sample = pd.DataFrame() N = 1000 sex_ratio = [6,4] # male : female education_ration = [3,3,4] # senior : high school : university age_range = [18,30] # range of age for ls in ['Num','Sex','Age','Education']: locals()[ls] = [] # Generate data for i in range(0,N): tmp = random.randint(1,sum(sex_ratio)) if tmp <= sex_ratio[0]: tmp_Sex = '男' else: tmp_Sex = '女' if tmp <= education_ration[0]: tmp_Education = '小学' if tmp in range(education_ration[0]+1,education_ration[0]+education_ration[1]+1): tmp_Education = '高中' if tmp in range(education_ration[0]+education_ration[1]+1, education_ration[0]+education_ration[1]+education_ration[2]+1): tmp_Education = '大学' tmp_Age = random.randint(age_range[0], age_range[1]) for ls in ['Sex', 'Age', 'Education']: locals()[ls].append(locals()[f'tmp_{ls}']) Num.append(i+1) for ls in ['Num','Sex','Age','Education']: sample[ls] = locals()[ls] sample.to_csv('sample.csv',index=None)
本回答被题主选为最佳回答 , 对您是否有帮助呢?解决 无用评论 打赏 举报
悬赏问题
- ¥15 metadata提取的PDF元数据,如何转换为一个Excel
- ¥15 关于arduino编程toCharArray()函数的使用
- ¥100 vc++混合CEF采用CLR方式编译报错
- ¥15 coze 的插件输入飞书多维表格 app_token 后一直显示错误,如何解决?
- ¥15 vite+vue3+plyr播放本地public文件夹下视频无法加载
- ¥15 c#逐行读取txt文本,但是每一行里面数据之间空格数量不同
- ¥50 如何openEuler 22.03上安装配置drbd
- ¥20 ING91680C BLE5.3 芯片怎么实现串口收发数据
- ¥15 无线连接树莓派,无法执行update,如何解决?(相关搜索:软件下载)
- ¥15 Windows11, backspace, enter, space键失灵