0

0

如何在不使用 NumPy 的情况下获取矩阵上三角区域(不含主对角线)的行列索引

心靈之曲

心靈之曲

发布时间:2026-02-02 11:03:13

|

226人浏览过

|

来源于php中文网

原创

如何在不使用 NumPy 的情况下获取矩阵上三角区域(不含主对角线)的行列索引

本文介绍一种纯 python 实现方式,用于获取二维方阵或矩形矩阵中上三角区域(严格上三角,即排除主对角线)所有元素的行索引与列索引,无需依赖 numpy。

在数值计算和矩阵操作中,常需遍历或提取矩阵的上三角部分(如计算相关系数矩阵、构造对称图邻接表等)。虽然 numpy.triu_indices() 可一键返回行、列索引元组,但在嵌入式环境、轻量级脚本或受限依赖场景下,我们往往需要纯 Python 解法。

核心思路是:对第 i 行,其上三角有效列索引为 i+1, i+2, ..., n-1(假设矩阵为 n×n 方阵;若为矩形,取 min(n, m) 作为列上限更稳妥)。我们逐行生成这些索引,并分别收集行号与列号。

以下为完整、健壮的实现(支持非方阵,自动适配实际列数):

def upper_triangle_indices(matrix):
    """
    返回严格上三角区域(不含主对角线)的 (row_indices, col_indices) 元组。
    支持二维列表(list of lists),不要求为方阵。
    """
    if not matrix or not matrix[0]:
        return [], []

    n_rows = len(matrix)
    n_cols = len(matrix[0])  # 以首行为基准,建议确保每行长度一致

    row_indices = []
    col_indices = []

    for i in range(n_rows):
        # 当前行的有效列范围:从 i+1 开始,但不超过 n_cols - 1
        start_col = i + 1
        end_col = min(n_cols, start_col + (n_cols - start_col))  # 等价于 min(n_cols, n_cols)
        # 更简洁写法:
        valid_cols = range(start_col, n_cols)

        row_indices.extend([i] * len(valid_cols))
        col_indices.extend(valid_cols)

    return row_indices, col_indices

# 示例使用
matrix = [[1, 2, 3],
           [4, 5, 6],
           [7, 8, 9]]

rows, cols = upper_triangle_indices(matrix)
print("行索引:", rows)  # [0, 0, 1]
print("列索引:", cols)  # [1, 2, 2]
print("对应元素:", [matrix[i][j] for i, j in zip(rows, cols)])  # [2, 3, 6]

关键说明

且听
且听

一个AI时代深度讲解分析好书的语音听库

下载
  • 该方法不依赖任何外部库,仅使用内置 range 和列表操作;
  • 自动兼容矩形矩阵(例如 4×3 矩阵中,第 2 行(i=2)起始列为 3,已超出列数,故无索引生成);
  • 返回两个平行列表,可直接用于后续索引访问(如 zip(rows, cols) 构造坐标对);
  • 时间复杂度为 O(n²),空间复杂度 O(k),其中 k 为上三角元素个数(≈ n²/2)。

⚠️ 注意事项

  • 输入矩阵应为“规则”二维列表(各行长度一致),否则可能引发 IndexError;可在函数开头添加校验(如 all(len(row) == n_cols for row in matrix));
  • 若需包含主对角线,将 start_col = i + 1 改为 start_col = i 即可;
  • 对于大型矩阵,纯 Python 循环性能低于 NumPy 向量化操作,但本方案胜在通用性与可移植性。

掌握此技巧,你便能在无 NumPy 环境中灵活操控矩阵结构——无论是构建稀疏索引、实现自定义矩阵运算,还是调试算法逻辑,都多了一把趁手的“纯 Python 解剖刀”。

热门AI工具

更多
DeepSeek
DeepSeek

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

豆包大模型
豆包大模型

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

通义千问
通义千问

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

腾讯元宝
腾讯元宝

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

文心一言
文心一言

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

讯飞写作
讯飞写作

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

即梦AI
即梦AI

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

ChatGPT
ChatGPT

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

相关专题

更多
页面置换算法
页面置换算法

页面置换算法是操作系统中用来决定在内存中哪些页面应该被换出以便为新的页面提供空间的算法。本专题为大家提供页面置换算法的相关文章,大家可以免费体验。

422

2023.08.14

AO3官网入口与中文阅读设置 AO3网页版使用与访问
AO3官网入口与中文阅读设置 AO3网页版使用与访问

本专题围绕 Archive of Our Own(AO3)官网入口展开,系统整理 AO3 最新可用官网地址、网页版访问方式、正确打开链接的方法,并详细讲解 AO3 中文界面设置、阅读语言切换及基础使用流程,帮助用户稳定访问 AO3 官网,高效完成中文阅读与作品浏览。

1

2026.02.02

主流快递单号查询入口 实时物流进度一站式追踪专题
主流快递单号查询入口 实时物流进度一站式追踪专题

本专题聚合极兔快递、京东快递、中通快递、圆通快递、韵达快递等主流物流平台的单号查询与运单追踪内容,重点解决单号查询、手机号查物流、官网入口直达、包裹进度实时追踪等高频问题,帮助用户快速获取最新物流状态,提升查件效率与使用体验。

0

2026.02.02

Golang WebAssembly(WASM)开发入门
Golang WebAssembly(WASM)开发入门

本专题系统讲解 Golang 在 WebAssembly(WASM)开发中的实践方法,涵盖 WASM 基础原理、Go 编译到 WASM 的流程、与 JavaScript 的交互方式、性能与体积优化,以及典型应用场景(如前端计算、跨平台模块)。帮助开发者掌握 Go 在新一代 Web 技术栈中的应用能力。

1

2026.02.02

PHP Swoole 高性能服务开发
PHP Swoole 高性能服务开发

本专题聚焦 PHP Swoole 扩展在高性能服务端开发中的应用,系统讲解协程模型、异步IO、TCP/HTTP/WebSocket服务器、进程与任务管理、常驻内存架构设计。通过实战案例,帮助开发者掌握 使用 PHP 构建高并发、低延迟服务端应用的工程化能力。

0

2026.02.02

Java JNI 与本地代码交互实战
Java JNI 与本地代码交互实战

本专题系统讲解 Java 通过 JNI 调用 C/C++ 本地代码的核心机制,涵盖 JNI 基本原理、数据类型映射、内存管理、异常处理、性能优化策略以及典型应用场景(如高性能计算、底层库封装)。通过实战示例,帮助开发者掌握 Java 与本地代码混合开发的完整流程。

1

2026.02.02

go语言 注释编码
go语言 注释编码

本专题整合了go语言注释、注释规范等等内容,阅读专题下面的文章了解更多详细内容。

61

2026.01.31

go语言 math包
go语言 math包

本专题整合了go语言math包相关内容,阅读专题下面的文章了解更多详细内容。

52

2026.01.31

go语言输入函数
go语言输入函数

本专题整合了go语言输入相关教程内容,阅读专题下面的文章了解更多详细内容。

25

2026.01.31

热门下载

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

精品课程

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

共4课时 | 22.4万人学习

Django 教程
Django 教程

共28课时 | 3.8万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.4万人学习

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

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