我正在尝试从一个csv文件复制多个列到另一个csv文件。列由通过命令行参数发送的参数指定。我让代码工作,这样它就可以复制在代码中静态设置的列,但是复制哪些列的决定需要是动态的。我的意思是,用户可以通过命令行指定他们想要的任何列,并且只复制这些列。我当前的代码如下,它复制了csv文件的前两列。我需要以某种方式遍历cmdargs并复制列表cmdargs中指定的行。有什么想法吗?在
import sys
import csv
import os
#Get rid of first argument (filename)
cmdargs = sys.argv[1:]
#Convert values to integers
cmdargs = list(map(int, cmdargs))
with open(r'data.csv','rb') as source:
rdr= csv.reader( source )
with open('output.csv',"wb") as result:
wtr= csv.writer( result )
for row in rdr:
wtr.writerow( ([row[0]], [row[1]]))
像这样的方法应该可以做到:
在
for row in rdr
内:或者如果你喜欢简短的版本:
^{pr2}$相关问题 更多 >
编程相关推荐