Python - 如何在 Pandas 中将 DataFrame 行分组到列表中?
pythonserver side programmingprogramming更新于 2023/10/8 6:56:00
要将 DataFrame 行分组到列表中，请使用 apply() 函数。首先，让我们导入所需库 −
import pandas as pd
创建包含 2 列的 DataFrame −
dataFrame = pd.DataFrame(
{
"Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],"Units": [100, 150, 110, 80, 110, 90]
}
)
使用 apply(list) 将 DataFrame 分组到列表中 −
dataFrame = dataFrame.groupby('Car')['Units'].apply(list)
示例
以下是代码 −
import pandas as pd
# 创建 DataFrame
dataFrame = pd.DataFrame(
{
"Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
"Units": [100, 150, 110, 80, 110, 90]
}
)
print"DataFrame ...\n",dataFrame
# 将数据框分组到列表中
dataFrame = dataFrame.groupby('Car')['Units'].apply(list)
# 显示数据框列表
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
