求可将Matlab切片与索引转换为Numpy数组表示的脚本

可以用Python结合正则表达式写一个简单的转换脚本，批量处理Matlab的索引语法，自动完成括号替换和索引减1的操作。以下是具体实现：

核心脚本代码 #

import re

def convert_matlab_index(match):
    # 提取括号内的索引部分
    index_part = match.group(1)
    
    def adjust_single_index(idx_str):
        # 纯冒号直接返回
        if idx_str == ':':
            return idx_str
        # 处理带冒号的范围（比如"2:45"、":12"）
        segments = idx_str.split(':')
        adjusted_segments = []
        for seg in segments:
            if seg.strip() == '':
                adjusted_segments.append('')
            else:
                adjusted_segments.append(str(int(seg) - 1))
        return ':'.join(adjusted_segments)
    
    # 按逗号分割多维度索引，逐个调整后合并
    adjusted_indices = ','.join([adjust_single_index(part.strip()) for part in index_part.split(',')])
    # 替换为Numpy的方括号格式
    return f'[{adjusted_indices}]'

def convert_matlab_code(matlab_code):
    # 匹配Matlab变量索引格式：变量名+括号包裹的索引
    pattern = r'([A-Za-z_]\w*)\((.*?)\)'
    return re.sub(pattern, convert_matlab_index, matlab_code)

# 测试示例
if __name__ == '__main__':
    test_code = """
    M(2:45,13:18)
    M(:12,:)
    V(4)
    V(:27)
    """
    converted_code = convert_matlab_code(test_code)
    print(converted_code)

脚本使用说明 #

1. 基础转换：运行脚本后，测试代码会输出转换后的Numpy切片：

   M[1:44,12:17]
    M[:11,:]
    V[3]
    V[:26]
   

2. 批量处理文件：如果要转换整个Matlab代码文件，添加以下代码：

   # 读取Matlab代码文件
   with open('input.m', 'r', encoding='utf-8') as f:
       matlab_code = f.read()
   # 执行转换
   python_code = convert_matlab_code(matlab_code)
   # 保存转换后的Python代码
   with open('output.py', 'w', encoding='utf-8') as f:
       f.write(python_code)
   

注意事项 #

• 脚本仅匹配变量(索引)格式的代码，若Matlab代码中有函数调用（比如plot(x,y)），需要手动排除这类场景，或扩展正则规则区分索引和函数调用
• 对于包含变量、表达式的复杂索引（比如M(a+2:b-1)），脚本无法直接处理，需要额外扩展逻辑支持表达式解析

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