python - 如何过滤制表符分隔的文本文件，该文件选择以特定字符串开头的行并转换为 CSV

Question

我有一个以制表符分隔的文本文件，我只想选择以特定字符串开头的行。然后我想将这些行转换为 CSV 文件。我能够做到这一点，但在 excel csv 中，文本文件中的每一行都被分成 csv 中一行的 3 个单元格，并且每个单元格内仍然有选项卡。此外，它还会跳过每隔一行。

我尝试用逗号替换制表符，但没有成功。

#parse APT.txt for airport data
import pandas as pd
import csv
import itertools

airport_data = source

APT_lines = []

for line in open(airport_data):
    if line.startswith('APT'):
        APT_lines.append(line)

df = pd.DataFrame(APT_lines)
df.to_csv('apt.csv', header=False, index=False, quoting=csv.QUOTE_NONE, escapechar=' ')

Answer 1

python 中的 csv 模块处理制表符分隔的文件以及逗号分隔的值。我想你想做这样的事情:

import csv
with open(input_file,newline='') as csvfile, open(output_file, 'w+', newline='') as output:
    reader = csv.reader(csvfile,delimiter='\t',quotechar='"')
    writer = csv.writer(output, delimited=',',quotechar='"')
    for row in reader:
        if len(row) != 0 and row[0].startswith('APT'):
            writer.writerow(row)

(我还没有测试过这段代码，您可能会发现其中有拼写错误；但是 CSV 模块使用起来很愉快。我建议直接将文件读取为 CSV，然后使用具有所需设置的 CSV 模块来将其写回。)