从2个数据帧（熊猫）中提取N个值

Question

我有2个具有相同列[ id ， value ]的熊猫数据框。

我想创建一个新的数据框，为第一个df1中的每个实例提取200个 id 值，如果没有足够的值，我应该转到第二个df2获取剩余的值。

我该怎么办？谢谢

Answer 1

您可以连接两个数据框并使用groupby().head()提取值。

(pd.concat((df1, df2))
  .groupby('id').head(200)
)