0

0

numpy 如何高效计算协方差矩阵但只用上三角

舞姬之光

舞姬之光

发布时间:2026-01-16 19:29:03

|

522人浏览过

|

来源于php中文网

原创

更高效计算协方差矩阵上三角:用np.triu_indices获取索引,中心化后向量化计算;或用np.einsum一次性求解;返回索引与值可节省内存,查(i,j)时利用对称性,分母为n-1确保无偏估计。

numpy 如何高效计算协方差矩阵但只用上三角

直接用 np.cov 计算完整协方差矩阵再取上三角,会做冗余计算和存储;更高效的方式是只计算上三角元素(含对角线),避免重复运算和多余内存占用

np.triu_indices 配合向量化计算

先获取上三角索引,再对每对变量(i, j)计算协方差,利用广播和向量化避免显式循环:

import numpy as np
<p>def cov_upper_tri(X):</p><div class="aritcle_card flexRow">
                                                        <div class="artcardd flexRow">
                                                                <a class="aritcle_card_img" href="/ai/2638" title="Woy AI"><img
                                                                                src="https://img.php.cn/upload/ai_manual/001/246/273/6972000e2bd59887.png" alt="Woy AI"  onerror="this.onerror='';this.src='/static/lhimages/moren/morentu.png'" ></a>
                                                                <div class="aritcle_card_info flexColumn">
                                                                        <a href="/ai/2638" title="Woy AI">Woy AI</a>
                                                                        <p>通过 Woy.ai AI 导航站发现 2024 年顶尖的 AI 工具!</p>
                                                                </div>
                                                                <a href="/ai/2638" title="Woy AI" class="aritcle_card_btn flexRow flexcenter"><b></b><span>下载</span> </a>
                                                        </div>
                                                </div><h1>X: (n_samples, n_features)</h1><pre class='brush:python;toolbar:false;'>n = X.shape[1]
iu = np.triu_indices(n)  # 上三角索引对 (row_idx, col_idx)
cov_vals = np.empty(len(iu[0]))

X_centered = X - X.mean(axis=0)  # 中心化一次,复用
for k, (i, j) in enumerate(zip(*iu)):
    cov_vals[k] = (X_centered[:, i] @ X_centered[:, j]) / (X.shape[0] - 1)

return iu, cov_vals

示例

X = np.random.randn(1000, 5) iu, vals = cov_upper_tri(X)

np.einsum 一次性计算上三角

更紧凑、更少 Python 循环,适合中等维度(如特征数 ≤ 100):

def cov_upper_tri_einsum(X):
    Xc = X - X.mean(axis=0)
    n = X.shape[0]
    C = np.einsum('ni,nj->ij', Xc, Xc) / (n - 1)
    iu = np.triu_indices_from(C)
    return iu, C[iu]
<h1>返回上三角索引和对应值,不构造全矩阵也可直接用</h1>

只存上三角 —— 节省内存的关键

若后续只需查询或遍历上三角元素,不必还原为二维数组。可封装为轻量结构:

  • iu(两个长度为 n*(n+1)//2 的整数数组)记录位置
  • 用一维数组 vals 存协方差值,顺序与 np.triu_indices 一致
  • 查 (i,j) 协方差:若 i np.searchsorted 或预建映射字典;若 i > j,利用对称性返回 vals[map[(j,i)]]

注意数值稳定性与自由度

协方差分母用 n_samples - 1(样本协方差),不是 n_samples;中心化务必沿 axis=0 进行,且推荐用 X - X.mean(0) 而非 scipy.stats.zscore 等额外开销操作。对超大样本,可考虑分块中心化避免内存峰值。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
golang map内存释放
golang map内存释放

本专题整合了golang map内存相关教程,阅读专题下面的文章了解更多相关内容。

77

2025.09.05

golang map相关教程
golang map相关教程

本专题整合了golang map相关教程,阅读专题下面的文章了解更多详细内容。

40

2025.11.16

golang map原理
golang map原理

本专题整合了golang map相关内容,阅读专题下面的文章了解更多详细内容。

67

2025.11.17

java判断map相关教程
java判断map相关教程

本专题整合了java判断map相关教程,阅读专题下面的文章了解更多详细内容。

47

2025.11.27

JavaScript浏览器渲染机制与前端性能优化实践
JavaScript浏览器渲染机制与前端性能优化实践

本专题围绕 JavaScript 在浏览器中的执行与渲染机制展开,系统讲解 DOM 构建、CSSOM 解析、重排与重绘原理,以及关键渲染路径优化方法。内容涵盖事件循环机制、异步任务调度、资源加载优化、代码拆分与懒加载等性能优化策略。通过真实前端项目案例,帮助开发者理解浏览器底层工作原理,并掌握提升网页加载速度与交互体验的实用技巧。

28

2026.03.06

Rust内存安全机制与所有权模型深度实践
Rust内存安全机制与所有权模型深度实践

本专题围绕 Rust 语言核心特性展开,深入讲解所有权机制、借用规则、生命周期管理以及智能指针等关键概念。通过系统级开发案例,分析内存安全保障原理与零成本抽象优势,并结合并发场景讲解 Send 与 Sync 特性实现机制。帮助开发者真正理解 Rust 的设计哲学,掌握在高性能与安全性并重场景中的工程实践能力。

68

2026.03.05

PHP高性能API设计与Laravel服务架构实践
PHP高性能API设计与Laravel服务架构实践

本专题围绕 PHP 在现代 Web 后端开发中的高性能实践展开,重点讲解基于 Laravel 框架构建可扩展 API 服务的核心方法。内容涵盖路由与中间件机制、服务容器与依赖注入、接口版本管理、缓存策略设计以及队列异步处理方案。同时结合高并发场景,深入分析性能瓶颈定位与优化思路,帮助开发者构建稳定、高效、易维护的 PHP 后端服务体系。

164

2026.03.04

AI安装教程大全
AI安装教程大全

2026最全AI工具安装教程专题:包含各版本AI绘图、AI视频、智能办公软件的本地化部署手册。全篇零基础友好,附带最新模型下载地址、一键安装脚本及常见报错修复方案。每日更新,收藏这一篇就够了,让AI安装不再报错!

84

2026.03.04

Swift iOS架构设计与MVVM模式实战
Swift iOS架构设计与MVVM模式实战

本专题聚焦 Swift 在 iOS 应用架构设计中的实践,系统讲解 MVVM 模式的核心思想、数据绑定机制、模块拆分策略以及组件化开发方法。内容涵盖网络层封装、状态管理、依赖注入与性能优化技巧。通过完整项目案例,帮助开发者构建结构清晰、可维护性强的 iOS 应用架构体系。

114

2026.03.03

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.5万人学习

Django 教程
Django 教程

共28课时 | 4.8万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.8万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号