python读取命令行参数 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:]的缘故是命令行第一个参数为文件名本身，这里没有意义