python - 用户访问任何 csv 文件的 csv 命令行参数

Question

我希望能够读取我笔记本电脑上的任何 csv 文件，而不仅仅是那个特定的文件。如何修复我的代码以要求用户在命令行上输入输入文件名？

ifile = open('esxtopresult.csv', 'rb')
reader = csv.reader(ifile)
ofile = open('esxtopresult_mod2.csv', 'wb')
writer = csv.writer(ofile)

标签： pythoncsv

Answer 1

简单地说，你可以像这样得到你所有的论点：

import sys
print sys.argv[1]
ifile = open(sys.argv[1], 'rb')

writer = csv.writer(ofile)


>>python code.py test/test.csv