daitianxie 2019-10-27 13:36 采纳率: 40%
浏览 2038
已采纳

python学习生成测试数据

各位大牛,我想生成按照一定的比例生成测试数据,比如我想生成1000条人员数据,其中的属性包括性别,学历,年龄等。其中男女比例为6:4,学历上初中、高中和大学的比例为3:3:4。
想生成这样的数据不知道我该怎么实现,恳请各位大佬指点一下,最好是用python实现,十分感激。

  • 写回答

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)
    
    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(1条)

报告相同问题?

悬赏问题

  • ¥250 雷电模拟器内存穿透、寻基址和特征码的教学
  • ¥200 比特币ord程序wallet_constructor.rs文件支持一次性铸造1000个代币,并将它们分配到40个UTXO上(每个UTXO上分配25个代币),并设置找零地址
  • ¥15 关于Java的学习问题
  • ¥15 如何使用chatgpt完成文本分类任务?
  • ¥15 已知速度v关于位置s的等式,怎么转化为已知位置求速度v的等式
  • ¥15 我有个餐饮系统,用wampserver把环境配置好了,但是后端的网页却进去,是为什么,能不能帮远程一下?
  • ¥15 R运行没有名称为"species"的插槽对于此对象类"SDMmodelCV"
  • ¥20 基于决策树的数字信号处理,2ask 2psk 2fsk的代码,检查下报错的原因
  • ¥15 wincc已组态的变量过多
  • ¥60 如图:直线与椭圆X轴平行,求直线与椭圆任意一点的相切坐标计算公式