如何为函数的多个变量提供多个参数

Question

我正在尝试通过 argparse 将多个参数用于函数中 我希望将参数放在下面的函数中

def toto(var1,var2,var3):
    print(var1,var2,var3)

这是我的参数解析

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('--names', type=str,nargs=1, help='<Required> Set flag')
args = parser.parse_args()    

问题是我需要三个参数。

我把函数放在像hello.py这样的文件中

if __name__ == "__main__":
    print(args)
    if args.names:
        toto(var1,var2,var3)

然后我运行命令

python hello.py --names 'bobo'

我有这个错误

Traceback (most recent call last):
  File "hello.py", line 15, in <module>
    toto(var1,var2,var3)
NameError: names 'var1' is not defined

发生了什么？如何添加三个参数而不是一个？

谢谢

Answer 1

首先，您需要在命令行上指定argparse 应接受的参数数量，在本例中为三个nargs=3。

parser.add_argument('--names', type=str,nargs=3, help='<Required> Set flag')

变量将存储在args.names，在这种情况下是一个列表，因此您可以将该列表解压缩为三个变量：

if __name__ == "__main__":
    if args.names:
        var1, var2, var3 = args.names
        toto(var1, var2, var3)

最后，像下面这样运行它，它会给你正确的结果：

~$ python hello.py --names 'babi' 'bibo' 'bobu'
babi bibo bobu