python - numpy loadtxt 和 savetxt 多个文件？

Question

我希望读取、计算并打印出多个相同格式的文件。

filenames2 = ["AAA", "BBB", "CCC", "DDD", "EEE"]
for filename2 in filenames2:
   with loadtxt (filename2, float) as data:
      a1 = data[:,0]
      b1 = data[:,3]
      c1 = data[:,4]
      d1 = data[:,5]
      e1 = data[:,6]
     %% DO something with all those data. 
   z = numpy.array((calculated_a11, calculated_b11, calculated_c11), dtype = float)
   z2 = z.T
   numpy.savetxt('avegedoutput_%s.txt' %filenames2, z2.reshape((1000,3)), fmt='%i %f %f')

我希望这样写，以便希望能够处理多个文件。但似乎 numpy savetxt 和 loadtxt 不能以这种方式工作......欢迎任何建议。

提前致谢。

ps)另外，我是否需要为这些数据读取部分制作空矩阵？似乎这些命令会产生以下错误:“TypeError:‘文件’对象不可订阅”。

我认为我需要制作零矩阵，并将数据放入这些矩阵中，就像在 Matlab 中所做的那样。这是正确的吗？

Answer 1

numpy.loadtxt接受一个文件名(字符串)并返回一个 numpy 数组。所以你不需要 with 子句:

filenames2 = ["AAA", "BBB", "CCC", "DDD", "EEE"]
for filename2 in filenames2:
    data = np.loadtxt(filename2)
    a1 = data[:,0]
    # ...
    np.savetxt('output_filename.txt', z, ...)

到目前为止，我认为没有必要在代码中创建零矩阵。 data 将与文件内容一起归档，a1、b1...将从 data< 获取其内容.