python学习生成测试数据

Question

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

Answer 1

先上代码，在‘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)