对一堆文件进行平均

Question

我有 1000 个文件，每个文件有 6 列 1000 分的信息。我想在 1000 个时间步中为每个点计算第 6 列的平均值。我的意思是对于每个点，我希望平均有 1000 个时间步，然后保存在一个文本文件中。我写的代码在理解过程中存在问题。它可以计算每个点的平均值，但最后它给了我一个数字。

import os
import numpy as np

l = [f for f in sorted(os.listdir('.')) if f.startswith('config')]


maxnum = np.max([int(os.path.splitext(f)[0].split('_')[1]) for f in l])
l = ['configuration_%d.out' % i for i in range(maxnum)]


for i, d in enumerate(l): 
    a = np.loadtxt(d).T 
    num = os.path.splitext(d)[0]
    #j = np.zeros(1001)

    x = np.mean(a[5])

    #y = np.mean(a[4])

    np.savetxt("foo.txt",x)

如果您能帮我一个忙，我将不胜感激。谢谢大家

Answer 1

回答您的问题，您每次都在覆盖 foo.txt，因此您只能看到一个数字。

在循环之前打开一个txt文件，然后添加信息：

txt_file = open('foo.txt', 'w')
for i, d in enumerate(l):
     txt_file.write(f'{x}\n')
txt_file.close()

两件事：

1. f"" 它是一个 f 字符串，可从 python3.6 使用。如果您使用的是旧版本，只需根据需要格式化字符串。
2. \n 表示换行，所以你把每个数字写在下一行。