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)