从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])

source.csv

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

results I got:

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  

对于如何去掉逗号有什么想法吗?

Answer 1

你需要给writerow()一个列表:

wtr.writerow([r[1]])

writerow()期望一个序列，一个有序的序列。列表是序列，但字符串也是;字符串可以被索引、迭代等等。通过传入r[1]，您给.writerow()一个单独字符的序列，每个字符写成一个单独的列。