使用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()

预期输出

3
6

实际输出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()