Question

我正在尝试从.dat文件中读取一些模拟结果并进行分析。这些文件具有以下结构：

 N o d a l   D i s p l a c e m e n t s     Time       0.10000E+01
                                            Prop. Ld.  1.00000E+00

Node     1 Coord     2 Coord     3 Coord     1 Displ     2 Displ     3 Displ
   1  0.0000E+00  5.0000E-01  0.0000E+00  0.0000E+00  0.0000E+00  0.0000E+00

   2  0.0000E+00  2.5005E-01  0.0000E+00  0.0000E+00  0.0000E+00  0.0000E+00

   3  0.0000E+00  1.0000E-04  0.0000E+00  0.0000E+00  0.0000E+00  0.0000E+00
   4  1.0000E+00  5.0000E-01  0.0000E+00 -1.9511E-04  4.0845E-04 -2.1522E-05

   5  1.0000E+00  2.5005E-01  0.0000E+00  1.1185E-08  4.0053E-04  2.6545E-09

   6  1.0000E+00  1.0000E-04  0.0000E+00  1.9511E-04  4.0847E-04  2.1526E-05

   7  2.0000E+00  5.0000E-01  0.0000E+00 -3.5177E-04  1.5287E-03 -1.2678E-05

......等等。

如何将数值数据转换为float并删除字符串？我尝试使用以下代码：

class DataLoader:
    def __init__(self, number_files):
        self.number_files = number_files

    def loader(self):
        array = []

        for i in range(1, self.number_files + 1):
            try:
                if i < 10:
                    data = open("ndis_00%s.dat" % i, "r")
                elif i >= 10 and i <= 100:
                    data = open("ndis_0%s.dat" % i, "r")
            except IOError:
                print("I/O Error")

            for line in data:
                if line != "\n":
                    array.append(line.split())
                    try:
                        float(line)
                    except (ValueError, TypeError):
                        line[:] = []
                    print(line)

在代码中我想保留转换为float的行，并且可以完全删除不转换的字符串。

Answer 1

如果文件的结构如上所述，则包含所需数据的行是非空的，并且不以字母字符开头。您可以按如下方式利用它：

def extractData(f):
    data = []
    for line in f:
        line = line.strip()
        if len(line) == 0 or line[0].isalpha(): continue
        items = line.split()
        data.append([float(item) for item in items[1:]])
    return data

上面的函数接受这样一个文件并返回一个包含你感兴趣的浮点数据的二维Python数组（列表列表）。

要测试（test.dat包含您的示例数据）：

with open('test.dat') as testf:
    nums = extractData(testf)
    for row in nums: print(row)

输出：

[0.0, 0.5, 0.0, 0.0, 0.0, 0.0]
[0.0, 0.25005, 0.0, 0.0, 0.0, 0.0]
[0.0, 0.0001, 0.0, 0.0, 0.0, 0.0]
[1.0, 0.5, 0.0, -0.00019511, 0.00040845, -2.1522e-05]
[1.0, 0.25005, 0.0, 1.1185e-08, 0.00040053, 2.6545e-09]
[1.0, 0.0001, 0.0, 0.00019511, 0.00040847, 2.1526e-05]
[2.0, 0.5, 0.0, -0.00035177, 0.0015287, -1.2678e-05]

如果这不起作用 - 您还没有充分定义文件的结构。

将字符串数组转换为float数组

1 个答案: