从特定内容读取文本文件和写入文件

Question

我想从文本文件创建和写入新文件。 我面临的挑战是如何获得特定栏目的内容。你知道吗

输入文件搞乱了。你知道吗

例如

input.txt（制表符分隔：行之间的列总数不同）

CATEGORY      NEIGHBOUR        NUMBER1   NUMBER2     TOTAL
city          Washington    30        50          80
county        mountain      in        seattle     10        4         30
community     church        men       15          5         4

output.txt（我想创建的内容。（每行3列）

CATEGORY                      NUMBER1      TOTAL    
city Washington               30           80
county mountain in seattle    10           30 
community church men          15           4

我该怎么写才能这样？你知道吗

Answer 1

尝试以下操作：

lines = open('in_file', 'r').readlines()
for line in lines:
    fields = line.split('\t')
    ...

Answer 2

您可以尝试以下python脚本：

file_in  = open('input.txt', 'r')
file_out = open('output.txt', 'w')

for line in file_in:
    line = line.rstrip()
    line = line.split('\t')

    CATEGORY = ' '.join(line[:-3]) ##set variable as beginning of line 
                                   ##to before 3rd last column
    NUMBER_1 = line[-3] ##set variable as 3rd last column
    TOTAL = line[-1]    ##set variable as last column

    file_out.write('%s\t%s\t%s\n' % (CATEGORY, NUMBER_1, TOTAL))

file_in.close()
file_out.close()

• 运行脚本依据：python <ABOVE_SCRIPT>.py（假设输入文件与脚本位于同一目录中）
• 输出将在输出.txt。。。你可以把它抹掉