命令行运行Python脚本时传入参数

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

Thanks to:

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

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

来源：helloasimo