在Python中读取文件并将列放入Array

时间:2013-11-22 23:39:24

标签: python arrays file csv

我是Python的新手,我正在使用NLTK生成一个声音动态文本分析器。我有一个.csv文件,其中包含我需要打开和阅读的成员信息,调查回复编号和调查回复文本。

我有:

import csv
import codecs

f = open('testresponseFS.csv')
raw = f.read()
print raw

这可能有点过头,但我想读取文件中的每一行以保持所有信息的完整性,并读取包含文本响应的特定单元格“响应”。我被建议将该特定列放在一个数组中,用数组值遍历整个列;因此,我可以在该数组中的每个项目上运行函数,并最终将这些值附加回“响应”旁边的.csv文件。

1 个答案:

答案 0 :(得分:3)

import csv

# read data
with open('testresponseFS.csv', 'rb') as inf:
    incsv = csv.reader(inf)
    header = next(incsv)
    data = [row for row in incsv]

# process data
header.append('Comments')
response_column = 4
for row in data:
    response = row[response_column]
    newval = response[:4].lower()    # or whatever you do to it
    row.append(newval)   

# write data back out
with open('finaldata.csv', 'wb') as outf:
    outcsv = csv.writer(outf)
    outcsv.writerow(header)
    outcsv.writerows(data)