我有多个具有相似名称的文件(只更改后缀)。我想在新文件中汇总文件的值。例如:文件energy.dat.1
包含矩阵格式的值,文件energy.dat.2
具有相同的格式,但具有其他值。
例如:
energy.dat.1
文件constains ([[1, 2], [3, 4]])
energy.dat.2
文件constains ([[0, 1], [1, 3]])
我想获得另一个total_energy
,其中包含总和的结果(即:([[1, 3], [4, 7]])
)。
因为我有很多文件:
for file_name in os.listdir('.'):
if file_name[0:10] == "energy.dat":
total_energy = total_energy +
np.sum(
np.loadtxt(file_name)
)
(之前我使用正确的维度定义了数组total_energy
。)
问题:结果total_energy
是一个矩阵,其值等于所有元素的总和文件(([[13, 13], [13, 13]])
)代替(([[1, 3], [4, 7]])
)
您有什么想法可以达到我想要的结果吗?
感谢