在Python中读取csv文件然后编写新的csv？

Question

我想创建一个简单的脚本来读取csv文件，然后根据列将所有值存储在数组中。

我的文件如下：

Date, Store name, UPC, Price, Cost
2014-05-26, Toronto Store, 10000078, 2.99, 1.20
2014-05-26, Markham Store, 10000079, 3.25, 1.50
...etc

我希望将日期存储在数组dates[]中，名称存储在storenames[]中等等。然后，我希望获取此信息并在操作后将其粘贴到新的csv文件中。

到目前为止，这就是我的全部内容：

import csv

cr = csv.reader(open("test.csv","rb"))
cw = csv.writer(open("newfile.csv", "wb"))
cw.writerow(["Date","Store Name","UPC","Price","Cost"])

Answer 1

python CSV module reference可能会有所帮助

像

这样的代码

import csv

csvfile = open('test.csv', 'rb')
your_array = []
reader = csv.reader(csvfile, delimiter=',')
for row in reader:
    your_array += row

应该按照自己的意愿行事，将所有值放入“your_array”。

如果你想按行和按字索引数组，你可以使用can change

your_array += row

到

your_array.append(row)

并按

索引

your_array[line][position]

Answer 2

查看DictReader和DictWriter课程。它们使用文件第一行中的名称将每个后续行转换为字典。然后，您可以通过访问原始列名来创建新列。对于csv＆＃34;中的所有行，你仍然需要做某种形式的＆＃34;处理，但逻辑应该很容易。