汇总python

时间:2016-05-24 12:17:17

标签: python

我有多个具有相似名称的文件(只更改后缀)。我想在新文件中汇总文件的值。例如:文件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]])

您有什么想法可以达到我想要的结果吗?

感谢

0 个答案:

没有答案