如何应用数组实现稀疏矩阵压缩算法实战降低无效零变量的内存占用

云浩酱_5047

云浩酱_5047

2026-05-13

169人浏览

原创

稀疏矩阵应选用CSR或CSC格式而非COO,批量构建避免逐个赋值,警惕隐式零与显式零混淆,合理指定dtype并用npz压缩存储以节省内存和提升性能。

直接用普通二维数组存大量零值矩阵,等于拿金库装几颗螺丝钉——空间浪费严重,计算也拖慢。稀疏矩阵压缩的核心,就是只记“谁不为零、在哪、值多少”,跳过所有零。

选对存储格式,别让格式拖垮性能

不是所有稀疏格式都适合大规模计算:

  • COO(坐标格式)适合初始化:用三个数组分别存行号、列号、值,构造直观,但不支持高效运算,也不支持原地修改;
  • CSR(按行压缩)更适合大多数场景:行访问快、矩阵乘法和迭代求解器(如共轭梯度法)天然适配;
  • CSC(按列压缩)更适配列操作:比如特征提取、sklearn 的 fit/transform 流程,或需要频繁取某一列的场景。

SciPy 中推荐直接用 sparse.csr_matrix() 或 sparse.csc_matrix() 构造,它们内部已做结构优化,索引和算术操作都是向量化实现。

批量构建,拒绝逐个赋值

写 mat[i, j] = x 看似自然,实则每次都会触发结构检查、内存重分配。100 万次赋值可能卡死或爆内存。

Marble
Marble

Marble是一款AI 3D生成工具,李飞飞联合创立的 World Labs 推出的 3D 世界生成平台。

下载
  • 先用 Python 列表收集所有非零项:rows, cols, data = [], [], [],循环中追加 (i, j, val);
  • 一次性构造:sparse.coo_matrix((data, (rows, cols)), shape=(M, N));
  • 若需进一步提速,可预分配 CSR 的 indptr 数组(尤其当每行非零元数量已知时);
  • 从文件加载时,优先用 sparse.load_npz() 读取已压缩的 .npz 文件,比现场解析快 5–10 倍。

警惕隐式零与显式零混淆

稀疏矩阵里“没存”不等于“不存在”。mat[0, 0] 返回 0 是逻辑结果,不是内存里真有个 0。

  • 调用 mat.toarray() 或 np.array(mat) 会强制转成稠密数组——10⁶×10⁶ 矩阵瞬间 OOM;
  • mat == 0 会返回稠密布尔矩阵,应改用 mat.count_nonzero() 或 mat.sum() 替代;
  • 用 np.where(mat) 提取非零位置前,确保 mat 已是 CSR/CSC 格式,否则 COO 下效率极低;
  • 长期运行后调用 mat.eliminate_zeros() 清理浮点误差引入的“近零值”和重复索引。

配合 dtype 和压缩策略再减负

默认 float64 占 8 字节,但多数场景 float32 或 int32 完全够用:

  • 创建时显式指定 dtype=np.float32,内存直降 50%,科学计算精度通常无损;
  • 计数类矩阵(如 TF-IDF、共现表)用 dtype=np.int32 更省;
  • 保存时用 sparse.save_npz("mat.npz", mat),比 pickle 小 3–5 倍,且跨平台可读。

相关文章

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

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

下载

相关标签:

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

相关专题

更多
java
java

Java是一个通用术语,用于表示Java软件及其组件,包括“Java运行时环境 (JRE)”、“Java虚拟机 (JVM)”以及“插件”。php中文网还为大家带了Java相关下载资源、相关课程以及相关文章等内容,供大家免费下载使用。

2023.06.15

12137

6

java正则表达式语法
java正则表达式语法

java正则表达式语法是一种模式匹配工具,它非常有用,可以在处理文本和字符串时快速地查找、替换、验证和提取特定的模式和数据。本专题提供java正则表达式语法的相关文章、下载和专题,供大家免费下载体验。

2023.07.05

9242

9

java自学难吗
java自学难吗

Java自学并不难。Java语言相对于其他一些编程语言而言,有着较为简洁和易读的语法,本专题为大家提供java自学难吗相关的文章,大家可以免费体验。

2023.07.31

7892

8

java配置jdk环境变量
java配置jdk环境变量

Java是一种广泛使用的高级编程语言,用于开发各种类型的应用程序。为了能够在计算机上正确运行和编译Java代码,需要正确配置Java Development Kit(JDK)环境变量。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.08.01

1284

3

java保留两位小数
java保留两位小数

Java是一种广泛应用于编程领域的高级编程语言。在Java中,保留两位小数是指在进行数值计算或输出时,限制小数部分只有两位有效数字,并将多余的位数进行四舍五入或截取。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.08.02

1028

3

java基本数据类型
java基本数据类型

java基本数据类型有:1、byte;2、short;3、int;4、long;5、float;6、double;7、char;8、boolean。本专题为大家提供java基本数据类型的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.02

1536

5

java有什么用
java有什么用

java可以开发应用程序、移动应用、Web应用、企业级应用、嵌入式系统等方面。本专题为大家提供java有什么用的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.02

3169

5

java在线网站
java在线网站

Java在线网站是指提供Java编程学习、实践和交流平台的网络服务。近年来,随着Java语言在软件开发领域的广泛应用,越来越多的人对Java编程感兴趣,并希望能够通过在线网站来学习和提高自己的Java编程技能。php中文网给大家带来了相关的视频、教程以及文章,欢迎大家前来学习阅读和下载。

2023.08.03

20431

3

配置java环境变量
配置java环境变量

配置Java环境变量是为了让操作系统能够识别和使用Java的相关命令和功能。本专题为大家提供配置java环境变量相关文章,帮助大家解决问题。

2023.08.03

1395

8

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Docker Desktop设置与维护手册
Docker Desktop设置与维护手册

共0课时 | 0人学习

Visual Studio 性能优化指南
Visual Studio 性能优化指南

共0课时 | 0人学习

ThinkPHP8.0完全开发手册
ThinkPHP8.0完全开发手册

共0课时 | 0人学习