如何从Python中的命令行参数获取文件名

时间:2015-04-29 11:12:33

标签: python input output

我有以下代码: 打开(" test.txt"," r")作为测试,打开(" table.txt"," w")作为表:     reader = csv.reader(test,delimiter =" \ t")     writer = csv.writer(table,delimiter =" \ t")     读者行:         if all(字段不在行中字段的关键字中):             writer.writerow(行) 我怎样才能将它转换为.py文件,让你定义table.txt并在运行时进行测试。所以必须写: Script.py test.txt output.txt

1 个答案:

答案 0 :(得分:1)

只需使用sys.argv

import sys
import csv

with open(sys.argv[1], "r") as test, open(sys.argv[2], "w") as table:
      # more here

注意,sys.argv[0]包含脚本名称(在您的情况下为Script.py)。要得到第一个参数,你应该得到sys.argv[1];要获得第二个参数,您应该获得sys.argv[2]等等。