Python 如何选择具有特定id的元素的随机样本

Python 如何选择具有特定id的元素的随机样本,python,pandas,Python,Pandas,我试图从一个包含审查ID和业务ID的数据框架中为每个业务选择5个审查ID。我以为它会在groupby对象上迭代,但这不起作用 df_b = df.groupby('business_id') selected = random.sample(df_b,5).review_ids 不确定DataFrameGroupBy对象是否是您想要的。我认为最好对每个id使用DataFrame.sample() frames = list() for buisness_id in df['business_i

我试图从一个包含审查ID和业务ID的数据框架中为每个业务选择5个审查ID。我以为它会在groupby对象上迭代,但这不起作用

df_b = df.groupby('business_id')
selected = random.sample(df_b,5).review_ids

不确定
DataFrameGroupBy
对象是否是您想要的。我认为最好对每个id使用
DataFrame.sample()

frames = list()
for buisness_id in df['business_id'].unique():
    frame = df[df['business_id'] == business_id].sample(5)
    frames.append(frame)

df = pd.concat(frames)

您可能希望给出一个示例数据帧,显示所需的输出,并解释代码如何不满足您的期望。