Python – 如何在Pandas中将DataFrame行分组为列表？

要将DataFrame行分组为列表，请使用apply()函数。首先，让我们导入所需的库 –

import pandas as pd

创建包含2列的DataFrame –

dataFrame = pd.DataFrame(
   {
      "汽车": ['宝马', '雷克萨斯', '奥迪', '野马', '宾利', '捷豹']，"销售量": [100, 150, 110, 80, 110, 90]
   }
)

使用apply（list）将DataFrame分组为列表 –

dataFrame = dataFrame.groupby('汽车')['销售量'].apply(list)

更多Pandas文章，请阅读：Pandas教程

示例

以下是示例代码 –

import pandas as pd

# 创建DataFrame
dataFrame = pd.DataFrame(
   {
      "汽车": ['宝马', '雷克萨斯', '奥迪', '野马', '宾利', '捷豹'],
      "销售量": [100, 150, 110, 80, 110, 90]
   }
)

print"DataFrame ...\n",dataFrame

#将dataframe分组成列表
dataFrame = dataFrame.groupby('汽车')['销售量'].apply(list)

# 显示dataframe列表
print"\nDataFrame ...\n",dataFrame

输出

这将产生以下输出 –

DataFrame ...
       汽车   销售量
0      宝马     100
1    雷克萨斯   150
2     奥迪     110
3    野马       80
4  宾利       110
5   捷豹        90

DataFrame ...
 汽车
奥迪       [110]
宝马       [100]
宾利       [110]
捷豹        [90]
雷克萨斯  [150]
野马        [80]
Name: 销售量, dtype: object