将Matlab二进制文件读取及重塑代码转换为Python的技术求助

关键差异说明 #

Matlab 的 fread 在指定 [M,N] 维度时，会按列优先（Fortran 顺序）将读取的元素填充为 M 行 N 列的矩阵；而 numpy 默认采用行优先（C 顺序）重塑数组，因此需要显式指定 order='F' 来匹配 Matlab 的行为。同时需要复刻原代码中的读取计数检查逻辑，确保读取到足够的数据。

修正后的Python代码 #

import numpy as np

binary_file = "你的二进制文件路径"

# 打开二进制文件（对应Matlab的fopen）
with open(binary_file, 'rb') as fid:
    # 读取前4个int32，对应Matlab的[inpar,ic]=fread(fid,4,'int')
    inpar = np.fromfile(fid, dtype=np.int32, count=4)
    if len(inpar) < 4:
        idata = np.array([])
        raise ValueError("读取inpar时数据不足")
    
    nmagic, nh, nrpar, nipar = inpar
    
    # 读取rdata：对应Matlab的[fread(binary_file,[nh,nrpar],'float')]
    rdata_count = nh * nrpar
    rdata_raw = np.fromfile(fid, dtype=np.float32, count=rdata_count)
    if len(rdata_raw) < rdata_count:
        raise ValueError("读取rdata时数据不足")
    # 按列优先重塑矩阵，匹配Matlab逻辑
    rdata = rdata_raw.reshape(nh, nrpar, order='F')
    
    # 读取idata：对应Matlab的[fread(fid,[nh,nipar],'int')]
    idata_count = nh * nipar
    idata_raw = np.fromfile(fid, dtype=np.int32, count=idata_count)
    if len(idata_raw) < idata_count:
        raise ValueError("读取idata时数据不足")
    idata = idata_raw.reshape(nh, nipar, order='F')

代码对应说明 #

1. 文件打开：使用 with open(...) 确保文件自动关闭，替代Matlab的fopen。
2. inpar读取：指定count=4读取固定数量的int32，检查长度是否达标，对应原代码的ic<4判断。
3. rdata/idata重塑：通过order='F'强制按列优先排列，完全对齐Matlab的矩阵填充逻辑。
4. 计数检查：通过判断读取到的数组长度是否等于预期数量，复刻原代码中ic<nh*nrpar的错误处理逻辑。

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