python获得命令行输入的参数的两种方式
外部直接执行python文件时,我们有时需要获得命令行的参数
获得命令行参数的两种方式
1、通过sys.argv
sys.argv:获得一个参数列表,第一个值为文件名本身,通过sys.argv[1]获得第文件名后的第一个参数 ,多个参数使用空格隔开
测试代码:
import sys print(sys.argv) print(len(sys.argv)) print(len(sys.argv[1]))
测试数据:
python3 test.py 第一个参数 第二个参数
执行结果:
['test.py', '第一个参数', '第二个参数']
3
5
[]
['第一个参数', '第二个参数']
2、通过getopt()获得参数值
如果命令行需要通过linux那中指定参数名的方法运行,可以使用getopt()来获得参数内容
命令【getopt.getopt(参数列表,"短选项参数",[长选项参数])】
参数有两种维度区分:短选项,长选项;带参数值(开关)和不带参数值
短选项参数(命令行使用-参数名):由1个字母组成,如果字母后面跟冒号:表示这个参数需要有参数值,否则表示不输入参数值
例如:io:
表示入参要输入-i -o=xxx
长选项参数(命令行使用--参数名):由[参数名=]表示,有值的参数一定要写=,否则会抛错
例如:["help","input=","output="]
表示入参要输入 --help --input=xxx --output=xxx
测试代码:
import getopt import sys opts,args=getopt.getopt(sys.argv[1:],"i:ho:",["help","input=","output="]) print(opts) print(args) for opts,arg in opts: print(opts) if opts=="-h" or opts=="--help": print("我只是一个说明文档") elif opts=="-i" or opts=="--input": print("我只是输入,输入内容如下:") print(arg) elif opts=="-o" or opts=="--output": print("我只是输出,输出内容如下:") print(arg)
测试数据:
python3 test.py -h -o=输出 --input=输入 最后一个参数
测试结果:
[('-h', ''), ('-o', '=输出'), ('--input', '输入')]
['最后一个参数']
-h
我只是一个说明文档
-o
我只是输出,输出内容如下:
=输出
--input
我只是输入,输入内容如下:
输入
说明:使用getopt()需要注意一下,不能把未定义的参数放到短选项和长选项的前面,否则-h --help这种不能读取到参数值(源码中不是很严谨,遇到这种情况直接就break了),放到后面可以,也能通过args获取到
getopt()中第一个参数写sys.argv[1:]的缘故是命令行第一个参数为文件名本身,这里没有意义
总结
下一篇:python Cartopy的基础使用详解