使用Pandas根据字典映射对DataFrame列进行求和-Python教程-PHP中文网

使用Pandas根据字典映射对DataFrame列进行求和

碧海醫心

发布： 2025-11-04 11:13:01

原创

141人浏览过

使用pandas根据字典映射对dataframe列进行求和

本文详细介绍了如何利用Pandas高效地根据一个字典映射，将DataFrame中的多列进行分组求和，生成一个新的DataFrame。我们将探讨两种主流的Pandas实现方法，包括利用rename结合groupby进行列操作，以及通过转置DataFrame来处理列分组求和，确保解决方案的简洁性、效率和对不同Pandas版本的兼容性。

在数据处理中，我们经常需要根据特定的业务逻辑或分析需求，将DataFrame中的现有列进行组合或聚合。一个常见的场景是，给定一个DataFrame和一套列名映射规则（通常以字典形式给出），我们需要创建新的列，其中每个新列是原始DataFrame中对应列的行级总和。本教程将展示如何以高度“Pandasic”的方式实现这一目标，避免使用低效的循环结构。

初始数据准备

首先，我们定义一个示例DataFrame和一个字典，该字典将新列名映射到原始DataFrame中需要求和的列名列表。

import pandas as pd

# 示例DataFrame
df = pd.DataFrame([[4,8,52,7,54],[0,20,2,21,35],[2,33,12,1,87]], 
                  columns = ['A', 'B', 'C', 'D', 'E'])

# 列名映射字典：键为新列名，值为原始列名列表
dic = {'x':['A','D'], 'y' : ['E'], 'z':['B','C']}

print("原始DataFrame:")
print(df)
print("\n列名映射字典:")
print(dic)

登录后复制

我们的目标是创建一个新的DataFrame，其中包含'x'、'y'、'z'三列。'x'列应是原始'A'和'D'列的行级和，'y'列是'E'列的行级和，'z'列是'B'和'C'列的行级和。

方法一：利用 rename 和 groupby(axis=1)

这种方法的核心思想是首先“反转”我们的映射字典，使其能够将原始列名映射到新的列名。然后，我们使用df.rename()函数根据这个反转后的字典临时修改列名，最后通过groupby(level=0, axis=1).sum()对具有相同新列名的列进行分组求和。

反转字典： 我们需要一个从原始列名到新列名的映射。例如，{'A': 'x', 'D': 'x', 'E': 'y', 'B': 'z', 'C': 'z'}。
```
d2 = {v:k for k,l in dic.items() for v in l}
print("\n反转后的字典:")
print(d2)
```
登录后复制
重命名并分组求和： 使用df.rename(columns=d2)将DataFrame的列名根据d2进行修改。此时，例如，原有的'A'和'D'列都会被临时命名为'x'。接着，groupby(level=0, axis=1).sum()会沿着列方向（axis=1）根据列名（level=0，因为列名是唯一的层级）进行分组，并对每个组内的列进行求和。
```
out_method1 = df.rename(columns=d2).groupby(level=0, axis=1).sum()
print("\n方法一结果:")
print(out_method1)
```
登录后复制
输出结果：

艺映AI
艺映AI - 免费AI视频创作工具

62

查看详情
```
    x   y   z
0  11  54  60
1  21  35  22
2   3  87  45
```
登录后复制

方法二：通过转置 DataFrame 进行处理

在较新版本的Pandas中，直接在groupby中使用axis=1可能会被弃用。为了兼容未来的版本或在某些情况下提供更清晰的逻辑，我们可以采用转置DataFrame的方法。这种方法将列转换为行，从而可以在行上执行groupby操作，这通常是groupby的默认行为，且更直观。

反转字典： 与方法一相同，我们首先需要一个从原始列名到新列名的映射d2。
```
# d2 已在方法一中创建
# d2 = {'A': 'x', 'D': 'x', 'E': 'y', 'B': 'z', 'C': 'z'}
```
登录后复制
转置、重命名、分组求和再转置：
- df.T：将DataFrame转置，使原始列名变为索引。
- .rename(d2)：此时，d2字典将用于重命名转置后的DataFrame的索引（即原始列名）。
- .groupby(level=0).sum()：在转置后的DataFrame上，根据索引（即新的列名）进行分组求和。
- .T：最后，将结果再次转置，恢复为我们期望的列式结构。
```
out_method2 = df.T.rename(d2).groupby(level=0).sum().T
print("\n方法二结果:")
print(out_method2)
```
登录后复制
输出结果：
```
    x   y   z
0  11  54  60
1  21  35  22
2   3  87  45
```
登录后复制

注意事项与总结

Pandas 版本兼容性： 方法一中的groupby(axis=1)在某些Pandas版本中可能会有弃用警告。方法二通过转置DataFrame来规避了这个问题，因此在长期维护和兼容性方面可能更具优势。
效率： 对于大型DataFrame，这两种Pandas的矢量化操作通常比手动循环迭代要高效得多。Pandas底层使用C语言实现，能够优化这些操作。
可读性： 两种方法都相对简洁，但方法二可能对初学者来说更易理解，因为它将列操作转换为更常见的行操作。
字典完整性： 确保dic字典中包含的原始列名都存在于DataFrame中，否则rename操作可能会引发KeyError或导致某些列不被处理。

通过这两种方法，我们可以优雅且高效地根据复杂的字典映射对DataFrame的列进行分组求和。选择哪种方法取决于您的Pandas版本、个人偏好以及对代码可读性的权衡。在实际应用中，理解并掌握这些高级的Pandas操作，能够显著提升数据处理的效率和代码质量。

以上就是使用Pandas根据字典映射对DataFrame列进行求和的详细内容，更多请关注php中文网其它相关文章！