将矩阵导出到文本文件时添加特殊标题的代码

import numpy as np

def export_matrix_to_text(matrix, filename, title):
    # 添加特殊标题
    header = f"=== {title} ===\n"

    # 将矩阵转换为字符串
    matrix_str = np.array2string(matrix, separator=', ')

    # 将标题和矩阵字符串写入文件
    with open(filename, 'w') as file:
        file.write(header)
        file.write(matrix_str)

# 示例用法
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
export_matrix_to_text(matrix, 'matrix.txt', '导出矩阵示例')