优化python代码

Question

我有以下代码：    inputFile = open('C:/Abaqus_JOBS' + JobDir + '/' + JobName + '-3_4.inp', 'r')    for line in inputFile:        fileData.append([x.strip() for x in line.split(',')])    fel=0    for row,data in enumerate(fileData):        if data[0]=='*Node':            row_nodes = row #number of the row when data='*Node'         if data[0]=='*Element' and fel==0:            row_elements2 = row            fel=1    for row,data in enumerate(fileData[row_nodes + 1:row_elements2]):        nodes.append(data) #data between '*Nodes' and '*Element'但是，它在外部程序的python交互程序中运行非常慢（几分钟）（我必须在这里运行脚本，因为我需要访问该程序产生的结果数据库）。我该如何优化呢？

Answer 1

如果我了解得很好，您首先要逐行存储文件，然后搜索“ * Element”的第一个匹配项和“ * Node”的最后一个匹配项，最后存储它们之间的内容。

我看到的一种优化是，您可以将文件的3次解析转换为单个解析：

inputFile = open('C:/Abaqus_JOBS' + JobDir + '/' + JobName + '-3_4.inp', 'r')

go_storage = False

nodes = None

for line in inputFile:

    if line[0] == "*Node":

        # Reset what has already been memorized

        nodes = list()

        go_storage = True

    elif line[0] == "*Element":

        break

    elif go_storage:

        nodes.append(line) 

Answer 2

也许您可以按照正则表达式来思考：

如果我理解正确，您想在某个文件中的关键词* Node和* Element之间获取数据，对吗？

好吧，您可以尝试类似：

import re

S = open('C:/Abaqus_JOBS' + JobDir + '/' + JobName + '-3_4.inp','r').read() 

Data =  re.finditer( "\*Nonde([.\n]*?)\*Element", S )

这应该为您提供在标签“ * Node”和“ * Elements”之间找到的字符串列表。

我希望那是您想要做的。