python - 向 argparse 参数添加附加参数

Question

我刚开始在 python 中使用 arparse 模块，希望有人能帮助我解决以下问题。我使用以下方法将可变数量的文件指定为输入:

parser = argparse.ArgumentParser(description='Get Files')    
parser.add_argument('-i','--input', help='Input file(s)',required=True, nargs='+') 
args = parser.parse_args()

我想指定可变数量的文件输入，每个文件输入的关联值为 1 或 2，但我不确定如何执行此操作。

我希望该程序能够运行，因此我的命令行条目应该是:

MyProgram.py -i myfile.txt 2 secondfile.txt 1 ...

一旦我开始工作，我该如何调用程序中的每个文件？

Answer 1

您的代码可以正常运行。你可以使用 grouper recipe一次循环遍历 args.input 两个项目:

import argparse
parser = argparse.ArgumentParser(description='Get Files')    
parser.add_argument('-i','--input', help='Input file(s)',required=True, nargs='+') 
args = parser.parse_args()
for filename, num in zip(*[iter(args.input)]*2):
    print(filename, num)
    # with open(filename) as f:
    #     ....

产量

('myfile.txt', '2')
('secondfile.txt', '1')