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

helloasimo

已于 2023-01-07 16:18:54 修改

阅读量5.3w

点赞数 41

分类专栏： python 文章标签： python

于 2022-04-16 11:00:48 首次发布

本文链接：https://blog.youkuaiyun.com/helloasimo/article/details/124210144

版权

python 专栏收录该内容

14 篇文章

订阅专栏

本文介绍了如何在Python中通过命令行传参，包括使用argparse模块解析命令行参数，并展示了sys.argv的用法。实例演示了如何根据参数条件执行不同操作，如打印、判断和动态输出。同时涵盖了如何获取脚本名称和处理不同数量的参数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

方法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）_于科技人文间徘徊的博客-优快云博客

Thanks to:

向python脚本传递参数_chengxuyuanyonghu的博客-优快云博客_python 参数传递

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