Python如何对不同文件中的多个矩阵数据做加法并输出指定格式结果

解决方案 #

核心需要调整numpy的默认输出规则，消除方括号、浮点数小数点和多余对齐空格，具体实现分两种常用场景：

场景1：直接在终端打印符合要求的结果

import numpy as np

data2 = np.loadtxt('20.dat', dtype='float')
data3 = np.loadtxt('21.dat', dtype='float')
# 求和后转为整数类型，消除浮点数小数点
sum_mat = (data2 + data3).astype(int)
# 逐行拼接为单空格分隔的字符串后输出
for row in sum_mat:
    print(' '.join(map(str, row)))

场景2：直接将结果保存为本地文件

不需要额外打印到终端的话可以直接用numpy内置方法输出，性能更高：

import numpy as np

data2 = np.loadtxt('20.dat', dtype='float')
data3 = np.loadtxt('21.dat', dtype='float')
sum_mat = (data2 + data3).astype(int)
# fmt='%d'指定输出整数，delimiter指定单空格分隔
np.savetxt('sum_result.dat', sum_mat, fmt='%d', delimiter=' ')

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