将硬编码文件名转换为命令行参数

Question

run.py 的脚本如下：

a = open('a.csv')
b = open('b.csv')
c = open('c.csv','w')

while True:
   la = a.readline()
   if not la: break
   lb = b.readline()
   la = la.split('\t')
   lb = lb.split('\t')
   la[4] = str(int(la[4])+int(lb[4]))
   la[5] = str(int(la[5])+int(lb[5]))
   c.write('\t'.join(la)); c.write('\n')

是否可以将其转换为以下格式：

python run.py a.csv b.csv c.csv

这样我就可以在Terminal中将文件名更改为参数，非常感谢。

Answer 1

您可以通过sys.argv访问传递给您的计划的参数。

from sys import argv

a = open(argv[1])
b = open(argv[2])
c = open(argv[3],'w')

# Etc.