如何在Python 3中读取对称Harwell-Boeing矩阵？

确实，SciPy的scipy.io.hb_read函数默认只支持读取非对称的Harwell-Boeing（HB）格式矩阵，不过有几种实用的办法可以处理对称HB矩阵，我之前解决这个问题时试过这些方案，都能正常工作：

方法1：修改HB文件头部的类型标记 #

HB格式的文件头部会标记矩阵类型，比如对称矩阵通常用S（对称）、H（Hermitian）或P（正定），而hb_read只识别A（非对称）类型。我们可以先读取文件内容，修改这个类型标记，再让hb_read处理：

from scipy.io import hb_read
from io import BytesIO
import numpy as np

def read_symmetric_hb(file_path):
    with open(file_path, 'rb') as f:
        content = f.read()
    # 将对称类型标记替换为非对称的'A'（仅替换第一次出现的标记）
    modified_content = content.replace(b'S', b'A', 1).replace(b'H', b'A', 1).replace(b'P', b'A', 1)
    # 用BytesIO模拟文件对象传入hb_read
    with BytesIO(modified_content) as f:
        tri_matrix = hb_read(f)
    # 恢复完整对称矩阵：避免对角线重复相加
    return tri_matrix + tri_matrix.T - np.diag(tri_matrix.diagonal())

注意：这个方法依赖标准HB格式的头部结构，大部分公开的HB文件类型标记位置都是固定的，适配性很强。

方法2：手动构造对称矩阵 #

如果不想修改文件内容，也可以直接用hb_read读取矩阵的三角部分（对称HB文件只会存储上/下三角元素），再手动填充对称部分：

import numpy as np
from scipy.io import hb_read

def read_symmetric_hb_manual(file_path):
    # 读取得到三角矩阵（可能是上三角或下三角，取决于文件存储方式）
    tri_matrix = hb_read(file_path)
    # 构造对称矩阵：对角线元素保留，非对角线元素取三角部分和其转置的和
    symmetric_matrix = tri_matrix + tri_matrix.T - np.diag(tri_matrix.diagonal())
    return symmetric_matrix

这个方法更稳妥，不需要依赖文件头部的修改，只要确认矩阵是对称存储的就能生效。

方法3：使用第三方库PyAMG #

PyAMG库的IO模块对HB格式的支持更全面，原生就能读取对称HB矩阵：

from pyamg.io import read_hb

# 直接读取得到对称矩阵
symmetric_matrix = read_hb('your_symmetric_matrix.hb')

注意：需要先安装PyAMG，执行pip install pyamg即可完成安装。

内容的提问来源于stack exchange，提问作者edub