Question

我想在我的python代码中做这样的事情

python my_prog.py -inp 3 -inp2 4

并且能够在我的python程序中使用inp和inp2作为输入。我该怎么做？

Answer 1

您正在寻找argparse模块。

import argparse

parser = argparse.ArgumentParser(description='Process some integers.')
parser.add_argument('input1', metavar='a', type=int, help='an input for foo')
parser.add_argument('input2', metavar='b', type=int, help='an input for bar')
args = parser.parse_args()

print(args.input1 + args.input2)

Answer 2

您可以使用getopt来解析输入参数。

来自纪录片的例子：

import getopt, sys

def main():
    try:
        opts, args = getopt.getopt(sys.argv[1:], "ho:v", ["help", "output="])
    except getopt.GetoptError as err:
        # print help information and exit:
        print str(err) # will print something like "option -a not recognized"
        usage()
        sys.exit(2)
    output = None
    verbose = False
    for o, a in opts:
        if o == "-v":
            verbose = True
        elif o in ("-h", "--help"):
            usage()
            sys.exit()
        elif o in ("-o", "--output"):
            output = a
        else:
            assert False, "unhandled option"
    # ...

if __name__ == "__main__":
    main()

在python程序中设置外部输入

2 个答案: