如何在从cmd提示符打开的Python脚本中输入变量？

Question

我想知道如何在从cmd提示符打开的Python脚本中输入变量？我知道在C语言中可以这样做：

int main( int argc, char **argv ) {
    int input1 = argv[ 0 ]
    int input2 = argv[ 1 ]

.....

}

我该如何在Python中实现相同的结果？

- Richard

---

在尝试读取input2之前，您应该检查参数的数量是否至少为2。 - HelloGoodbye

Answer 1

import sys

def main():
   input1 = sys.argv[1]
   input2 = sys.argv[2]
...

if __name__ == "__main__":
   main()

- froadie

Answer 2

参数在 sys.argv 中，第一个是脚本名 sys.argv[0]。

对于更复杂的参数解析，应使用 argparse （对于 Python >= 2.7）。以前用于此目的的模块为 getopts 和 optparse。

- Mad Scientist

Answer 3

有两个选择。

1. 导入 sys.argv 并使用它。
2. 使用 getopts

另请参阅: Dive into Python 和 PMotW

- Sean Vieira

Answer 4

确定选项特定变量也非常有用

''' \
USAGE:  python script.py -i1 input1 -i2 input2
    -i1 input1 : input1 variable
    -i2 input2 : input2 variable
'''

import sys 
...

in_arr = sys.argv
if '-i1' not in in_arr  or '-i2' not in in_arr:
    print (__doc__)
    raise NameError('error: input options are not provided')
else:
    inpu1 = in_arr[in_arr.index('-i1') + 1]
    inpu2 = in_arr[in_arr.index('-i2') + 1]
...

# python script.py -i1 Input1 -i2 Input2

- Trigremm