Pandas GroupBy 多重索引转换：从 Series 轻松还原至 DataFrame

python

2024-03-17 13:42:46

Pandas GroupBy 多重索引：从 Series 转换回 DataFrame

问题

如果你对 Pandas DataFrame 执行 GroupBy 操作，就会得到一个多重索引的 GroupBy 对象。但是，你可能希望将其转换回一个包含所有行的 DataFrame。

解决方案

将 Pandas GroupBy 多重索引输出从 Series 转换回 DataFrame 的步骤如下：

1. 重置索引

首先，使用 reset_index() 方法重置 GroupBy 对象的索引，将其转换为一个具有原始列名的 DataFrame：

df_reset = g1.reset_index()

2. 设置索引

然后，使用 set_index() 方法将所需的列设置为 DataFrame 的索引：

df_final = df_reset.set_index(['Name', 'City'])

示例

为了演示，我们创建一个示例 DataFrame：

import pandas as pd

df = pd.DataFrame({'City': ['Seattle', 'Seattle', 'Portland', 'Seattle', 'Seattle', 'Portland'],
                   'Name': ['Alice', 'Bob', 'Mallory', 'Mallory', 'Bob', 'Mallory']})

对其执行 GroupBy 操作：

g1 = df.groupby(['Name', 'City'])

然后，重置索引：

df_reset = g1.reset_index()

最后，将所需的列设置为索引：

df_final = df_reset.set_index(['Name', 'City'])

输出：

                  City  Name
Name    City
Alice   Seattle      1     1
Bob     Seattle      2     2
Mallory Portland     2     2
Mallory Seattle      1     1