[Python知识库]命令行运行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 命令行传参

创作挑战赛

新人创作奖励来咯，坚持创作打卡瓜分现金大奖