熊猫分层抽样

使用

min

df

df = pd.Dataframe(dict(        A=[1, 1, 1, 2, 2, 2, 2, 3, 4, 4],        B=range(10)    ))df.groupby('A', group_keys=False).apply(lambda x: x.sample(min(len(x), 2)))   A  B1  1  12  1  23  2  36  2  67  3  79  4  98  4  8