数据挖掘-分层抽样

#分层抽样
gbr = data.groupby("area")
 
gbr.groups
 
typicalFracDict = {
    1: 0.2, 
    2: 0.4, 
    3: 0.6
}
 
def typicalSampling(group, typicalFracDict):
    name = group.name
    frac = typicalFracDict[name]
    return group.sample(frac=frac)
 
result = data.groupby(
    'area', group_keys=False
).apply(typicalSampling, typicalFracDict)