在python中打开多个文件并将数据存储在数组中答案

【问题标题】：Open multiple files and store the data in arrays in python在python中打开多个文件并将数据存储在数组中
【发布时间】：2021-08-10 10:40:45
【问题描述】：

import numpy as np
import matplotlib.pyplot as plt
import csv

file = 'random_file'
with open(file + '.txt') as datafile:
    for line in range(7):
        next(datafile)
    for line in datafile:
        lines = datafile.readlines()
        x0_1  = [float(line.split()[0]) for line in lines]
        y1_1  = [float(line.split()[1]) for line in lines]
        Z_1   = [float(line.split()[2]) for line in lines]
        u_1   = [float(line.split()[3]) for line in lines]
        v_1   = [float(line.split()[4]) for line in lines]
        V0_1  = [float(line.split()[5]) for line in lines]
        V1_1  = [float(line.split()[6]) for line in lines]
        Yf_1  = [float(line.split()[7]) for line in lines]
        Yo_1  = [float(line.split()[8]) for line in lines]
        T_1   = [float(line.split()[9]) for line in lines]
        Yp0_1 = [float(line.split()[10]) for line in lines]
        Yp1_1 = [float(line.split()[11]) for line in lines]
        D_1   = [float(line.split()[12]) for line in lines]
        rho_1 = [float(line.split()[13]) for line in lines]
#Do something with the data

目前，我有这段代码可以打开单个文件，跳过文件中的前七行，然后将文件中的数据存储到数组中。但是我有多个文件，我想以类似的方式读取文件中的数据并将其存储到数组中。到目前为止，我只是为不同的文件重复上面的代码。如何一次读取所有文件并将数据存储在数组中？

数据文件看起来像这样，

#First seven lines ignore #

  0.00000E+00  5.00000E-04  9.28613E-01  6.58135E-02  4.05154E-01  6.58135E-02  4.05154E-01  9.24508E-01  2.01228E-16  4.44076E+02  1.12880E-02  9.23567E-03  2.46667E-05  4.53488E-01
  0.00000E+00  1.50000E-03  8.14544E-01  9.64630E-02  3.82685E-01  9.64630E-02  3.82685E-01  8.03881E-01  0.00000E+00  6.74294E+02  2.93252E-02  2.39933E-02  2.46667E-05  3.15181E-01
  0.00000E+00  2.50000E-03  5.46555E-01  1.40368E-01  1.23704E-01  1.40368E-01  1.23704E-01  5.20482E-01  0.00000E+00  1.21516E+03  7.17009E-02  5.86644E-02  2.46667E-05  2.01024E-01
  0.00000E+00  3.50000E-03  1.23207E-01  1.08260E-01 -2.56808E-01  1.08260E-01 -2.56808E-01  7.27916E-02  0.00000E+00  2.06958E+03  1.38643E-01  1.13435E-01  2.46667E-05  1.54495E-01

【问题讨论】：

将此代码放入一个函数中，并在您要读取的每个文件中调用它。此外，您可以使用zip 函数来“转置”表格，而不是一遍又一遍地编写[float(line.split()[0]) for line in lines]。

标签： python arrays file

【解决方案1】：

您可以创建一个包含所有文件名的数组并遍历它们。只要文件以相同的方式存储数据并且您想在其中读取相同的内容。

类似这样的：

files = ['file1', 'file2', 'file3']
    
for file in files:
  with open(file + '.txt') as datafile:
  # Skip the first seven lines, Read the rest of the file

【讨论】：

【解决方案2】：

应该是评论，因为它不是直接的答案，但在评论中格式化代码是不可能的。

注意：您的代码实际上跳过了 8 行，效率极低。问题出在这里：

for line in range(7):
    next(datafile)               # ok, 7 lines skipped
for line in datafile:            # read 8th line 
    lines = datafile.readlines() # lines only starts at the 9th line

此外，您为每个字段重复分割行，效率不高。

由于您已经导入了 csv 模块，您可以这样做：

for line in range(7):
    next(datafile)               # ok, 7 lines skipped
x0_1, y1_1, Z_1, u_1, v_1, V0_1, V1_1, Yf_1, Yo_1, T_1, Yp0_1, Yp1_1, D_1, rho_1 = [
    [float(i) for i in row] for row in zip(*csv.reader(datafile, delimiter=' ',
                                                       skipinitialspace=True))][:14]

【讨论】：