首頁 >後端開發 >Python教學 >如何將 Pandas DataFrame 行分組並將列值轉換為清單?

如何將 Pandas DataFrame 行分組並將列值轉換為清單?

Linda Hamilton
Linda Hamilton原創
2024-12-20 09:13:09658瀏覽

How to Group Pandas DataFrame Rows and Convert Column Values to Lists?

如何將Dataframe 行轉換為Pandas GroupBy 中的列表

在Pandas 中操作dataframe 時,可能需要將資料轉換為特定格式格式以供進一步分析。一種方法是按指定列對行進行分組,並從每個組中的另一列建立清單。

在這種情況下,我們得到一個包含兩列的資料框:「a」(列名稱)和'b'(列值)。任務是將此資料框轉換為新的資料框,其中「a」列中的每個唯一值都將「b」列中的對應值分組到一個清單中。

要實現此目的:

df1 = df.groupby('a')['b'].apply(list).reset_index(name='new')

在此程式碼中:

  • df.groupby('a'):按列將資料框分組'a'.
  • ['b'].apply(list):將列表函數應用於每個組,將'b'列值轉換為列表。
  • reset_index(name= 'new'): 重置結果資料幀的索引並將新列的名稱設為 'new'。

最終結果是一個新的資料框 df1,其中「a」列中包含「a」列中的唯一值,以及「new」列中「b」列中的對應清單。

這裡有一個範例來說明:

給定以下資料框:

a b
A 1
A 2
B 5
B 5
B 4
C 6

應用上述程式碼將對其進行轉換進入:

a new
A [1, 2]
B [5, 5, 4]
C [6]

以上是如何將 Pandas DataFrame 行分組並將列值轉換為清單?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn