使用 Python 从 CSV 文件中提取列数据

Question

任务：使用python解析一个CSV文件，并将一列的内容输出到一个文本文件中。

输入文件（in.csv）：

one,two,three  
four,five,six  

Python 脚本：

#!/usr/bin/env python

import csv

inputFile = open('in.csv', 'r')
inputReader = csv.reader(inputFile)

outputFile = open('out.txt', 'w')
outputWriter = csv.writer(outputFile)

for row in inputReader:
        text = row[2]
        # write column 3 to file
        outputWriter.writerow(text)

outputFile.close()
inputFile.close()

预期输出

三个
六个

实际输出

t,h,r,e,e
s,i,x

Answer 1

csvwriter.writerow() 方法需要一个列表。因此，它在列表上下文中使用text 中的字符串；即它使用text 作为字符列表，每列一个：

#!/usr/bin/env python

import csv

inputFile = open(name, 'r')
inputReader = csv.reader(inputFile)

outputFile = open('onetwothree.txt', 'w')
outputWriter = csv.writer(outputFile)

for row in inputReader:
        text = row[2]
        # write column 3 to file
        outputWriter.writerow([text])

outputFile.close()
inputFile.close()