如何在Python中根据特定的权重对DataFrame数据进行加权平均计算?

阿丽酱_4895

阿丽酱_4895

2026-07-24

1004人浏览

原创

pandas.DataFrame.dot是最直接实现列加权平均的方法,要求权重为索引匹配DataFrame列名的Series,且数据须全为数值类型;若需逐行加权,则推荐numpy.average配合axis=1和显式归一化的权重。

如何在python中根据特定的权重对dataframe数据进行加权平均计算?

用 pandas.DataFrame.dot 做加权平均最直接

如果你的权重是列向量(即一维数组或 Series),且权重长度与 DataFrame 列数一致,dot 是最简洁、最不易出错的方式。它本质是矩阵乘法,天然适配「每列乘对应权重再求和」的加权平均逻辑。

注意:这里默认你要求的是**按列加权后得到一个标量结果**(即对整张表做一次加权平均),而不是逐行加权。如果是逐行加权,请跳到下一个副标题。

  • 权重必须是 Series,索引需与 DataFrame 的 columns 完全匹配,否则会自动对齐——对齐失败的部分变成 NaN,导致整个结果为 NaN
  • 如果 DataFrame 含非数值列(如字符串),dot 会报 ValueError: matrices are not aligned;务必先用 select_dtypes('number') 过滤
  • 示例:
    import pandas as pd<br>df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})<br>weights = pd.Series([0.3, 0.7], index=['A', 'B'])<br>result = df.select_dtypes('number').dot(weights).mean()  # 先加权再取均值(若需单个标量)

逐行加权平均用 numpy.average 配合 axis=1

当你要对每一行单独计算加权平均(比如每条样本有自己的权重向量),numpy.average 比 Pandas 原生方法更可控,尤其支持 axis 和 weights 参数直出结果。

  • 输入必须是纯数值的二维数组,所以得先用 .values 或 .to_numpy() 转换;DataFrame 直接传入会触发隐式转换警告甚至错误
  • 权重必须是一维数组,长度等于列数;若给的是二维权重(每行权重不同),则必须确保 weights 形状与数据一致,并显式指定 axis=1
  • 缺失值(NaN)会导致结果全为 NaN,除非你提前用 np.nanmean 类函数处理,但 np.average 不支持自动跳过 NaN —— 此时应改用 pd.DataFrame.apply + 自定义函数
  • 示例:
    import numpy as np<br>arr = df.to_numpy()  # shape: (3, 2)<br>w = np.array([0.3, 0.7])  # 一行一个权重,广播生效<br>row_wavg = np.average(arr, weights=w, axis=1)  # shape: (3,)

权重归一化不是可选项,而是必要步骤

很多人忽略这点:如果权重和不为 1,np.average 和 df.dot 算出来的根本不是数学意义上的「加权平均」,而是加权和。Pandas 不会帮你归一化,NumPy 也不会。

macOS UI 自动化技能包 —— 用 Python 控制你的 Mac
macOS UI 自动化技能包 —— 用 Python 控制你的 Mac

让 macOS 自动化像说话一样简单 —— 用代码控制你的 Mac,无需学习 AppleScript 系统要求 macOS 14+ (Sonoma) Python 3.9+ 需要开启:辅助功能 + 屏幕录制权限

下载
  • 错误做法:np.average(x, weights=[2, 8]) → 实际等价于 (2*x0 + 8*x1)/10?不,是除以 2+8=10,但用户常误以为它自动归一了——其实它确实除以了权重和,但如果你本意是「权重代表概率」,就必须自己保证和为 1
  • 安全做法:始终显式归一化,例如 w_norm = w / w.sum(),再传入;或者用 weights=w/sum(w) 写死
  • 特别注意:当权重含负数或零时,归一化后仍可能产生非预期行为(如方差爆炸),此时加权平均本身已失去统计意义,应提前校验权重有效性

遇到 ValueError: shapes (n, m) and (k,) not aligned 怎么办

这是 dot 最常见的报错,核心就一条:列名没对上,或数据类型不一致。

  • 检查 df.columns.tolist() 和 weights.index.tolist() 是否完全相等(顺序、大小写、空格都算)
  • 检查 DataFrame 是否有重复列名:df.columns.duplicated().any() 为 True 时 dot 必炸
  • 检查权重是否为 Series:用 type(weights) 确认,list 或 ndarray 传进去会报这个错
  • 临时调试技巧:把权重转成字典再重建 Series,强制对齐:
    weights = pd.Series(dict(zip(df.columns, [0.3, 0.7])))

加权平均看着简单,真正卡住人的往往是权重和列之间的“对齐契约”——它不声不响,但错一点就全盘失效。

Python免费学习笔记(深入):立即使用
在学习笔记中,你将探索 Python 的核心概念和高级技巧!

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

2023.07.20

1851

4

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

2023.07.25

5004

7

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.31

1789

3

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

2023.08.03

30097

23

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3587

5

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3607

5

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

1243

5

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.10

616

4

python是前端还是后端
python是前端还是后端

Python属于前端也属于后端,其灵活性和丰富的生态系统使得开发人员能够在不同的领域中灵活运用。本专题为大家提供python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

2783

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程