Извлечь N значений из 2 кадров данных (pandas)

Question

у меня есть 2 pandas фреймов данных с такими же столбцами [ id , value ].

Я хочу создать новый фрейм данных, извлекающий 200 значений для каждого id , берущего экземпляры из первого df1, и если у него недостаточно значений, я должен go ко второму df2 взять оставшиеся значения .

Как я могу это сделать? Спасибо

Answer 1

Вы можете объединить два кадра данных и извлечь значения с помощью groupby().head().

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