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

Question

我有 2 个具有相同列 [ id , value ] 的 Pandas 数据框。

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

我能怎么做？ 谢谢

Answer 1

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

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