python - 如何从 csv 中读取没有标题的列并使用 Python 将输出保存在 txt 文件中？

Question

我有一个包含许多列的文件“TAB.csv”。我想从 CSV 文件中选择一列没有标题(该列的索引为 3)。然后创建一个新的文本文件“NEW.txt”并在其中写入该列(不带标题)。

下面的代码读取该列但带有标题。如何省略标题并将该列保存在新的文本文件中？

import csv
with open('TAB.csv','rb') as f:
    reader = csv.reader(f)
    for row in reader:
        print row[3]

Answer 1

这是@tmrlvi 所说的解决方案:它通过 next 函数跳过第一行(标题):

import csv

with open('TAB.csv','rb') as input_file:
    reader = csv.reader(input_file)
    output_file = open('output.csv','w')
    next(reader, None)

    for row in reader:
        row_str = row[3]
        output_file.write(row_str + '\n')

    output_file.close()