命令行运行Python脚本时传入参数--3种方法

方法1. 在命令行运行Python脚本时，传入参数，并根据参数做相应动作。

先来一个简单的---把命令行输入的参数打印出来。

Step1:  在linux中，建立名字为 t1.py 的python 脚本。t1.py 里面内容如下：

import argparse
if __name__ == "__main__":
  parser = argparse.ArgumentParser()
  parser.add_argument('--aa', type=str, default = None)
  parser.add_argument('--bb', type=int, default=32)
  args = parser.parse_args()
  print(args.aa)
  print(args.bb)

Step2: 在terminal中输入  python   t1.py   --aa=15  --bb=10

python   t1.py   --aa=15  --bb=10

结果如下： aa, bb都能正确打印。

再升级一下：

新建一个t2.py, 内容如下：

import argparse
if __name__ == "__main__":
  parser = argparse.ArgumentParser()
  parser.add_argument('--aa', type=str, default = None)
  parser.add_argument('--bb', type=int, default=32)
  args = parser.parse_args()
  if(args.bb==3):
    print(f'hello world, args.bb={args.bb}')
  else:
    print(f'sorry, args.bb is not 3, but ={args.bb}')

分别输入下面两条命令，看一下效果：

命令：

 python t2.py --aa=15 --bb=3
 python t2.py --aa=15 --bb=10

效果：

方法2. 使用sys.argv

Step1: 新建 script.py 这个文件，内容如下：

import sys
gpus = sys.argv[1]
batch_size = sys.argv[2]
print(gpus)
print(batch_size)

Step2: 命令行输入如下命令  

python     script.py    0,1,2    10

效果如下：

方法3.其他：把运行的python脚本名字打印出来。

Step1:将下面代码命名为t1.py

import sys
print(sys.argv[0])

Step2: 在terminal 下输入 python   t1.py

 会显示t1.py

再试验更多参数：

命令： python 4_sys_argv.py 1 2 3

其中4_sys_argv.py 内容如下：

import sys
print(sys.argv[0])
print(sys.argv[1])
print(sys.argv[2])
print(sys.argv[3])

结果：

如果出现下面错误：

Traceback (most recent call last):
  File "1_sys_arg.py", line 3, in <module>
    print(sys.argv[1])
IndexError: list index out of range

解决办法：

代码格式问题，每行结尾部分删除掉多余内容；

参考： IndexError: list index out of range 错误原理及解决方法（python）_于科技人文间徘徊的博客-CSDN博客

Thanks to:

向python脚本传递参数_chengxuyuanyonghu的博客-CSDN博客_python 参数传递

命令行运行Python脚本时传入参数的三种方式_Daniel2333的博客-CSDN博客_python 命令行传参