Python，如何提取包含特定字符的行

Question

输入：

=name Aa Ba Ca DD Ea
sldkclskdf
opkmnbv
=name Ab Bb Cb Db Eb
po,omome
nbnwnejkvjekw
=name Ac Bc Cc DD Ec
lkecvkkw
=name Ad Bd Cd Dd Ed
sdlkcmksldmksd
=name Ae Be Ce DD Ee
clskdjfs

输出：

=name Aa Ba Ca DD Ea
sldkclskdf
opkmnbv
=name Ac Bc Cc DD Ec
lkecvkkw
=name Ae Be Ce DD Ee
clskdjfs

我正在从文件中提取数据。我只想获取包含特定字符（即“DD”）的行以及带有“while 循环”的特定行之后的行信息。

with open(file, 'r') as fr, open(file_modified, 'w') as fw:

    temp = ''
    while(line):
        line = fr.readline()

        if line.startswith('='):
            fw.write(',' + temp + '\n') 

            templist = line.strip().split()
            for element in templist:
                if element.startswith('DD'):
                    fw.write(templist) 

            temp = ''
        else:
            temp += line.strip() 
     
    fw.write(temp)

Answer 1

你可以这样做：

INFILE = 'elold.txt'
OUTFILE = 'elnew.txt'
WRITELINE = False

with open(INFILE) as infile:
    with open(OUTFILE, 'w') as outfile:
        for line in infile:
            if line.startswith('='):
                WRITELINE = 'DD' in line
            if WRITELINE:
                outfile.write(line)