python调用随机分层抽样方法_如何使用Python进行随机分层抽样（不是训练/测试分割）？...

这是我目前为止最好的解决方案。重要的是，在每个地层之前对连续变量进行分类，并使观测值最少。在

在本例中，我是：产生人口

纯随机抽样

随机分层抽样

当比较两个样本时，分层样本更能代表总体人口。在

如果任何人有一个更好的方法，请随时分享。在import pandas as pd

import numpy as np

# Generate random population (100K)

population = pd.DataFrame(index=range(0,100000))

population['income'] = 0

population['income'].iloc[39000:80000] = 1

population['income'].iloc[80000:] = 2

population['sex'] = np.random.randint(0,2,100000)

population['age'] = np.random.randint(0,4,100000)

pop_count = population.groupby(['income', 'sex', 'age'])['income'].count()

# Random sampling (100 observations out of 100k)

random_sample = population.iloc[

np.random.randint(

0,

len(population),

int(len(population) / 1000)

)

]

# Random Stratified Sampling (100 observations out of 100k)

stratified_sample = list(map(lambda x : population[

(

population['income'] == pop_count.index[x][0]

)

&

(

population['sex'] == pop_count.index[x][1]

)

&

(

population['age'] == pop_count.index[x][2]

)

].sample(frac=0.001), range(len(pop_count))))

stratified_sample = pd.concat(stratified_sample)