我们在实际工作中,经常会遇到通过命令—— python ***.py 来运行python代码的场景,有时候我们还希望在命令行中传入参数,然后在代码中运行,解决这个问题就需要使用sys.argv。sys.argv[]就是一个从程序外部获取参数的桥梁,我们从外部取得的参数可以是多个,所以获得的是一个列表list,也就是说sys.argv其实可以看作是一个列表,所以才能用[]提取其中的元素。其第一个元素是程序本身,随后才依次是外部给予的参数。话不多说上代码!
import sys
a=sys.argv[1]
b=sys.argv[2]
c=sys.argv[0]
print(a)
print(b)
print(c)
以命令行的方式运行脚本,python pytest2.py p1 p2 这里p1和p2是两个参数,输出
p1
p2
pytest2.py
可以看到,我们在命令行中输入的参数p1和p2 成功的赋值给代码中的参数a和b。
当运行python pytest2.py p1,使用1个参数时,则会输出错误信息:
Traceback (most recent call last):
File "pytest2.py", line 3, in
b=sys.argv[2]
IndexError: list index out of range
原因很简单就是代码中我们输出sys.argv[2],而实际我们只传入了一个参数。
我们通过使用异常处理的方式,可以增强程序的健壮性,代码如下:
try:
a = sys.argv[1]
b = sys.argv[2]
c = sys.argv[0]
print(a)
print(b)
print(c)
except:
print("请输入两个参数")
以命令行的方式运行脚本python pytest2.py p1 p2 p3 p4,程序会正常运行,新增代码
print(sys.argv)
则会看到完整的输出
['pytest2.py', 'p1', 'p2', 'p3', 'p4']
讲到这里,大家就掌握了如何通过命令行的方式运行脚本并传入参数供py代码调用的正确使用姿势啦!原创不易,如果文章帮到了您,请帮忙点赞转发!