如何提取 Pandas DataFrame 中每組的第一行？

Pandas DataFrame：取得每個群組的第一行

在此上下文中，您有一個包含分組資料的pandas DataFrame，並希望從每個群組中提取第一行。這可以透過多種方法來實現。

一個簡單的方法是利用first()函數，它從每列檢索第一個非空值：

df.groupby('id').first()

此方法排除輸出中的索引列。要將id 列包含為列，請使用reset_index():

df.groupby('id').first().reset_index()

或者，您可以使用head(n) 取得每組的前n 行：

df.groupby('id').head(2).reset_index(drop=True)

在在此範例中， drop=True 參數會從輸出中刪除重設索引列。

另一種方法涉及在中設定 as_index 參數groupby():

df.groupby('id', as_index=False).nth(0)

此方法預設包含 id 作為一列。

請記住根據您對列包含的具體要求和要檢索的行數來調整這些方法.

以上是如何提取 Pandas DataFrame 中每組的第一行？的詳細內容。更多資訊請關注PHP中文網其他相關文章！