Question

我正在使用Windows 7.0并安装了Python 3.4。我是Python的新手。这是我的清单。这是一个价格文件。我有成千上万的这些，但一直试图让它只在一个工作。

我试图仅提取以hfus，ious或oaus开头的行。

caus    123456  99.872300000        2
gous    1234567 99.364200000        2
oaus    891011  97.224300000        2
ious    121314  96.172800000        2
hfus    151617  99081.00            2
hfus    181920  1.000000000         2

这是理想的结果。

oaus    891011  97.224300000        2
ious    121314  96.172800000        2
hfus    151617  99081.00            2
hfus    181920  1.000000000         2

这是我到目前为止所写的内容，但它不起作用。我也想如果它遍历每个文件并用截断的列表覆盖现有文件，用它的原始名称保存它。文件033117.txt表示日期。每个文件都保存为mmddyy.txt。让它在所有文件上工作将是理想的，但是现在如果我可以让它甚至可以在一个文件上工作那么好。

inFile = open("033117.txt")
outFile = open("result.txt", "w")
buffer = []
keepCurrentSet = True
for line in inFile:
    buffer.append(line)
    if line.startswith("hfus"):
        if line.startswith("oaus"):
            if line.startswith("ious"):
        if keepCurrentSet:
            outFile.write("".join(buffer))
        keepCurrentSet = True
        buffer = []
    elif line.startswith(""):
        keepCurrentSet = False
inFile.close()
outFile.close()

Answer 1

我建议您在打开文件对象时使用with语句，这样就不需要显式关闭文件，当退出缩进块时，它会自动关闭。从文件中读取和过滤并将结果写入另一个文件（不覆盖同一文件）可以通过使用list comprehension并选择合适的行来完成，这些行提供了一种更简洁的方式来完成任务：

with open("033117.txt", 'rt') as inputf, open("result.txt", 'wt') as outputf:    
    lines_to_write = [line for line in inputf if line.split()[0] in ("hfus", "ious", "oaus")]
    outputf.writelines(lines_to_write)

如果要覆盖文件而不是打开新的附加文件并写入，请执行以下操作：

with open('033117.txt', 'r+') as the_file: 
    lines_to_write = [line for line in the_file if line.split()[0] in ("hfus", "ious", "oaus")] 
    the_file.seek(0)  # just to be sure you start from the beginning (but it should without this...)  
    the_file.writelines(lines_to_write)
    the_file.truncate()

有关开放模式，请参阅open, modes。

Answer 2

with open('033117.txt') as inFile, open('result.txt', 'w') as outFile:
    for line in inFile:
        if line.split()[0] in ('hfus', 'ious', 'oaus'):
            outFile.write(line)

Answer 3

尝试此查询：

inFile = open("033117.txt")
outFile = open("result.txt", "w")
for line in inFile.readlines():
    if line.startswith("hfus"):
        outFile.write(line)
    if line.startswith("oaus"):
        outFile.write(line)
    if line.startswith("ious"):
        outFile.write(line)
inFile.close()
outFile.close()

即使是python的新手，所以可能有更多更好的解决方案，但这应该有效。

Answer 4

对于这种数据处理，我建议使用pandas

import pandas as pd
df = pd.read_csv("033117.txt", header=None, names=['foo','bar','foobar','barfoo'])
df = df[df.foo.isin(['hfus','oaus'])]
df.to_csv("result.txt")

当然，您希望使用更有意义的标头值; - ）

Answer 5

尝试使用with语句打开您的文件，而不是outFile = open()。这应该有助于减少错误：）

with open('033117.txt') as inFile, open('result.txt', 'w') as outFile:
    for line in inFile:
        if line.split()[0] in ('hfus', 'ious', 'oaus'):
            outFile.write(line)

从基于第一列的文本文件中提取行到Python中的文本

5 个答案: