python从文件中查找数据并输出

Question

如题所述

Answer 1

在处理文件时，我们可能需要搜索特定的数据。这里提供了一个Python脚本，用于从文件夹中的文件中查找包含关键词的行，并提取特定格式的数据。首先，确保安装了所需的库，例如re，用于正则表达式操作。

示例代码如下：

注意，这里的代码用单空格缩进

导入正则表达式库

import re

设置你的文件夹路径

yourdir=""

定义关键词

keywordA="keywordA"

keywordB="keywordA(\d+)"

获取文件夹中的文件列表

files=[os.path.join(yourdir,f)for fin os.listdir(yourdir)]

以写模式打开输出文件

with open("out.txt","w") as fo:

遍历文件列表

for fin files:

获取当前文件名

fname=os.path.basename(f)

以读模式打开文件

with open(f,"r") as fi:

遍历文件的每一行

for line in fi:

如果行不为空

if line.strip():

搜索关键词A

searchA=re.search(keywordA,line.strip())

如果找到关键词A

if searchA:

搜索关键词B

searchB=re.search(keywordB,line.strip())

如果找到关键词B

if searchB:

打印文件名和提取的数据到输出文件

print(fname,serachB.groups()[0],sep="\t",file=fo)

此脚本可以用来从文件中查找包含特定关键词的行，并提取额外的信息，如数字。输出结果会被保存到out.txt文件中。

为了确保代码正确运行，请确保已经安装了Python和相关库，并正确设置了文件夹路径和关键词。