使用 Python 从文本文件中提取特定行

Question

您好，请参阅之前已解决帖子的链接。

Copy specific lines from text files into excel

此 VBA 代码允许我打开文件夹中的所有文本文件并将特定行加载到 Excel 电子表格中。

但是，我现在正在处理更大的文件和更多的文件。使用 VBA 不再是一种选择，因为它花费的时间太长，而且数据限制如何。

我想知道是否有任何现有的 python 代码可以从每个数据文件中提取特定行，并将其保存为新文档或将自身保存为其当前文件名。

示例代码：

DATASET UNSTRUCTURED_GRID
POINTS 5 float
0.096853 0.000000 0.111997
0.096853 -0.003500 0.111997
0.096890 0.000000 0.084015
0.096853 -0.003500 0.111997
0.096890 -0.003500 0.084015
CELL_DATA 5
SCALARS pressure float 1
LOOKUP_TABLE default
-0.000000
-0.000000
-3.000000
-2.000000
-6.000000

对此的任何提示将不胜感激。谢谢， 乔恩

Answer 1

可以试试这个：

# use python 3.6
from pathlib import Path 

txt_file_content = """DATASET UNSTRUCTURED_GRID
POINTS 5 float
0.096853 0.000000 0.111997
0.096853 -0.003500 0.111997
0.096890 0.000000 0.084015
0.096853 -0.003500 0.111997
0.096890 -0.003500 0.084015
CELL_DATA 5
SCALARS pressure float 1
LOOKUP_TABLE default
-0.000000
-0.000000
-3.000000
-2.000000
-6.000000"""

# creating sample file
Path('sample.txt').write_text(txt_file_content)

上面的代码创建了一个示例文件，然后解析它：

# read a file back, itrate over many files if needed
doc = Path('sample.txt').read_text()

# NOTE:
# you can walk over *.txt files in specific fodler with 
# https://docs.python.org/3/library/glob.html#glob.glob


# assume the disired text block is 
#   (1) always after 'LOOKUP_TABLE default'
#   (2) at the end of txt file
last_text_segment = doc.split('LOOKUP_TABLE default')[1]

values = [float(x) for x in last_text_segment.split('\n') if x]

# alternatively as a function:

def extract_pressure(filename):
    doc = Path(filename).read_text()
    last_text_segment = doc.split('LOOKUP_TABLE default')[1]
    return [float(x) for x in last_text_segment.split('\n') if x]

您可能希望将数据组装到 pandas 数据帧中，以便对其进行进一步的数字运算。

Answer 2

试试这个：

counter=0
with open(filename, 'w') as infile:
    for line in infile:
            counter+=1
            if line.split(' ')[0]=='CELL_DATA':
            i=counter+3
            j=line.split(' ')[1]+1
            break

这部分找到您应该开始检索数据的行。然后你可以对数据做任何你喜欢的事情。例如：

data=[]
for line in infile[i:i+j]:
    data.append(line)

并将数据保存在某处只需使用 python 选项写入 excel 或任何其他文件。 祝你好运！