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

【问题标题】：Extract N values from 2 dataframes (pandas)从 2 个数据帧（熊猫）中提取 N 个值
【发布时间】：2020-06-16 01:19:45
【问题描述】：

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

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

我该怎么办？谢谢

【问题讨论】：

【解决方案1】：

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

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

【讨论】：