Question

我需要将参数传递给我的python程序，例如：

py program.py var1=value var2=value var3=value var4=value

def main(*args):
    variable1 = args[var1]
    ...

基本上，我需要能够将参数值赋给变量，但是，用户可以按任何顺序传入参数，因此我需要确保在正确的位置使用正确的参数值代码。

这可能吗？如果是这样，怎么样？

Answer 1

如果您想坚持使用当前格式，可以使用字典来完成此操作。

假设：

传入的每个变量都采用varname=value格式。

您知道用户应该放置哪些变量名称，这意味着您希望声明某些变量在您的程序中使用，并且您知道用户将给出的名称。

代码基本上如下所示：

from sys import argv

def main():
    #All possible variables the user could input
    parameter_dict = {}
    for user_input in argv[1:]: #Now we're going to iterate over argv[1:] (argv[0] is the program name)
        if "=" not in user_input: #Then skip this value because it doesn't have the varname=value format
            continue
        varname = user_input.split("=")[0] #Get what's left of the '='
        varvalue = user_input.split("=")[1] #Get what's right of the '='
        parameter_dict[varname] = varvalue

    #Now the dictionary has all the values passed in, and you can reference them by name, but you'll need to check if they're there.

    #Then to access a variable, you do it by name on the dictionary.
    #For example, to access var1, if the user defined it:
    if "var1" in parameter_dict:
        print("var1 was: " + parameter_dict["var1"])
    else: #Or if the user did not define var1 in their list:
        print("User did not give a value for var1")


main()

测试（文件名为test.py）：

$python3 test.py var2=Foo
User did not give a value for var1
$python3 test.py var1=Bar
var1 was: Olaf
$python3 test.py var7=Goose var3=Pig var1=Grape
var1 was: Grape
$python3 test.py hugo=victor var1=Napoleon figs=tasty
var1 was: Napoleon

这样做，您也可以迭代所有用户的输入，尽管词典不保留顺序：

#Inside main() after previous items    
    for varname in parameter_dict:
        print(varname + "=" + parameter_dict[varname])

输出：

$python3 test.py frying=pan Panel=Executive Yoke=Oxen
User did not give a value for var1
frying=pan
Yoke=Oxen
Panel=Executive

Answer 2

就个人而言，我会使用argparse.ArgumentParser。有关详细信息，请参阅https://docs.python.org/3/library/argparse.html。

import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--var1', action="store");
parser.add_argument('--var2', action="store");

args = parser.parse_args();
print("var1 = %s" % args.var1);
print("var2 = %s" % args.var2);

Answer 3

如果您想这样做，可以随时使用os.environ。

from os import environ

def main():
    variable1 = environ.get("var1", False)
    variable2 = environ.get("var2", False)
    print(variable1, variable2)

main()

然后你会var1=value1 var2=value2 python test.py传递值，而不是传递它们作为参数。这会将它们设置为环境变量，并执行从shell环境中获取变量的程序。此代码还使用.get()方法提供默认值（在本例中为False），因此您可以轻松检查变量是否存在。

Answer 4

您可能需要查看sys.argv。

在你的例子中......

py program.py var1=value var2=value var3=value var4=value

def main(*args):
    variable1 = args[var1]
    ....

argv[0] = program.py

argv[1] = var1=value

等

然后你可以用split() ...

解析它

variable1 = argv[1].split('=')[1]

如何通过sys.argv名称将变量值传递到python中的main？

4 个答案: