如何基于数据帧的列的值随机选择数据帧的记录，同时也有一些条件？

Question

我有这样一个数据集：

df

我想选择"MI"个样本并添加到另一个数据帧。例如：如果Ranking=0，我想随机选择MI=2行。或者如果Ranking=1，我想选择MI=5行。 这是我的代码，但不起作用

All_M=pd.DataFrame()
A= df['Ranking'].min()

for i in range(0 , len(df)):
    x6 = df[(df['Ranking'] == A)].apply(lambda x:x.sample(int(df["MI"][i])).reset_index(drop=True))
    All_M= x6.append(All_M)
    A = A + 1 

Answer 1

IIUC，我想你可以不用循环使用groupby语句：

 new_df = df.groupby('rank', group_keys=False)\
            .apply(lambda x: x.sample(x.iloc[0, x.columns.get_loc('MI')]))