csv - 从 csv 结果中删除列错误

Question

试图从 csv 文件中删除除“fruit”之外的所有列。

我从本网站的另一篇文章中修改的脚本

import csv
with open("C:\\Test\\source.csv","rb") as source:
    rdr= csv.reader(source)
    with open("C:\\Test\\result.csv","wb") as result:
        wtr= csv.writer(result)
        for r in rdr:
            wtr.writerow(r[1])

源文件.csv

ID,fruit,number,location  
0,apple,2,3  
1,banana,1,1  
2,pear,2,3  
3,grape,0,1  

我得到的结果：

f,r,u,i,t  
a,p,p,l,e  
b,a,n,a,n,a  
p,e,a,r  
g,r,a,p,e  

期望的结果

fruit  
apple  
banana  
pear  
grape  

关于如何摆脱逗号的任何想法？我正在使用python 2.7。

Answer 1

你需要给出writerow()一个清单：

wtr.writerow([r[1]])

writerow()需要一个序列，有序的东西。列表是序列，但字符串也是；可以对字符串进行索引、迭代等。通过传入，r[1]您只需给出.writerow()一系列单独的字符，每个字符都写为单独的列。