Python - 如何將 DataFrame 行分組到 Pandas 中的列表中？

要將資料框行分組到列表中，請使用 apply() 函式。首先，讓我們匯入所需庫 -

import pandas as pd

使用 2 列建立資料框 -

dataFrame = pd.DataFrame(
   {
      "Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],"Units": [100, 150, 110, 80, 110, 90]
   }
)

使用 apply(list) 將資料框分組到列表中 -

dataFrame = dataFrame.groupby('Car')['Units'].apply(list)

示例

以下是程式碼 -

import pandas as pd

# Create DataFrame
dataFrame = pd.DataFrame(
   {
      "Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
      "Units": [100, 150, 110, 80, 110, 90]
   }
)

print"DataFrame ...\n",dataFrame

# grouping dataframe into lists
dataFrame = dataFrame.groupby('Car')['Units'].apply(list)

# displaying dataframe lists
print"\nDataFrame ...\n",dataFrame

輸出

這將產生以下輸出 -

DataFrame ...
       Car   Units
0      BMW     100
1    Lexus     150
2     Audi     110
3  Mustang      80
4  Bentley     110
5   Jaguar      90

DataFrame ...
 Car
Audi      [110]
BMW       [100]
Bentley   [110]
Jaguar     [90]
Lexus     [150]
Mustang    [80]
Name: Units, dtype: object

更新時間：2021 年 9 月 20 日

297 次瀏覽

開啟您的 職業生涯

完成課程後獲得認證

開始
Advertisement 廣告