Python中如何将任意大小单位矩阵统一格式保存至CSV文件？

问题根源 #

你当前的代码直接将numpy数组对象传入csv.writerow，当矩阵尺寸达到84x84时，numpy的默认字符串格式化会自动添加换行符，csv.writer会保留这些换行符，导致CSV内容被拆分成多行，出现格式混乱。而83x83的矩阵因为元素总数未触发numpy的换行阈值，所以能保持单行格式。

解决方案 #

通过numpy的array2string方法手动控制字符串格式，强制生成无换行的矩阵字符串，再写入CSV：

import numpy as np
import csv

n = 84  # 可替换为任意大小
A = np.identity(n)

# 转换为无换行的单行字符串，保持矩阵格式
matrix_str = np.array2string(
    A,
    separator=', ',    # 元素间分隔符，与默认打印格式一致
    suppress_small=True,  # 将极小浮点数（如1e-16）显示为0
    threshold=np.inf      # 关闭自动换行阈值，强制单行输出
)

with open("output.csv", 'w') as f:
    writer = csv.writer(f)
    writer.writerow([matrix_str])  # 将字符串作为单个单元格写入

关键参数说明 #

• threshold=np.inf：告诉numpy不管数组多大，都不触发自动换行逻辑，确保输出为单行字符串。
• suppress_small=True：单位矩阵中对角线外的元素是极小的浮点数（因numpy浮点数精度问题），该参数会将其转为0，与直观的单位矩阵格式一致。

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