用户定义的输入csv文件

Question

我以某种格式读取csv文件的一些代码行.我现在想要的(并且无法找到解决方案)是文件名在调用python程序时停止过时并成为实际的用户输入.现在我在代码中有一个静态文件名,例如:

reader = csv.reader(open("file.csv", "rb"))

我希望这成为用户提供的信息.这是使用raw_input完成的还是通过让用户在以下行中编写内容来实现它:

python testfile.py --input:file.csv

提前感谢您的帮助!

Answer 1

最简单的方法是编写这样的脚本:

import sys
reader = csv.reader(open(sys.argv[1], "rb"))

然后像这样运行它:

python testfile.py file.csv

你应该进行一些错误检查,例如:

if len(sys.argv) < 2:
    print "Usage..."
    sys.exit(1)

要获得更多功能,请使用optparse/ argparse库中的bult .