1.argparse模块作用

用于解析命令行参数

2.位置参数和可选参数

运行以下代码：

import argparse

parser = argparse.ArgumentParser()

args = parser.parse_args()

运行结果：

参数为-h或--help时，parser.parse_args()会输出命令行的位置参数position arguments和可选参数optional arguments

位置参数：按照参数的顺序解析，参数必填

可选参数：以-或--开头，参数非必填

位置参数，可选参数示例如下：

import argparse

parser = argparse.ArgumentParser()

parser.add_argument("num", type=int, help="position")

parser.add_argument("--mode", "-m", type=int, help="optional")

args = parser.parse_args()

运行结果：

num为一个位置参数，type表示参数的类型，参数默认都是string类型的，help表示参数的描述

--mode，以--开头，为一个可选参数，可以有多个别名（--mode,-m）

可选参数，不是必填的，一般用来作条件分支，如下所示：

import argparse

parser = argparse.ArgumentParser()

parser.add_argument("num", type=int, help="position")

parser.add_argument("--mode", "-m", type=int, help="optional")

args = parser.parse_args()

result = 0

if args.mode == 1:

    result = args.num ** 2

else:

    result = args.num * 2

print(result)

运行结果：

当mode=1时，计算num的平方，否则，计算num*2

指定参数的范围,choice

import argparse

parser = argparse.ArgumentParser()

parser.add_argument("num", type=int, choices=[5, 6, 7], help="position")

parser.add_argument("--mode", "-m", action="store_true", help="optional")

args = parser.parse_args()

result = 0

if args.mode:

    result = args.num ** 2

else:

    result = args.num * 2

print(result)

运行结果：

3.可选参数

位置参数没有太多要讲的，必填，类型必须和定义的一致（默认为string），否则会报错。

可选参数有以下几种action

1)store_true

有参数时为true，没有为false，有参数时不能指定参数的值

import argparse

parser = argparse.ArgumentParser()

parser.add_argument("num", type=int, help="position")

parser.add_argument("--mode", "-m", action="store_true", help="optional")

args = parser.parse_args()

result = 0

if args.mode:

    result = args.num ** 2

else:

    result = args.num * 2

print(result)

运行结果：

2)count

数值，-m表示1，-mm或-m -m表示2，以此类推

import argparse

parser = argparse.ArgumentParser()

parser.add_argument("num", type=int, choices=[5, 6, 7], help="position")

parser.add_argument("--mode", "-m", action="count", help="optional")

args = parser.parse_args()

result = 0

if args.mode == 1:

    result = args.num ** 2

elif args.mode == 2:

    result = args.num * 2

else:

    result = args.num

print(result)

不加参数，表示None，可以设置默认count，则-m表示默认值+1

import argparse

parser = argparse.ArgumentParser()

parser.add_argument("num", type=int, help="position")

parser.add_argument("--mode", "-m", action="count", default=1, help="optional")

args = parser.parse_args()

result = 0

if args.mode == 1:

    result = args.num ** 2

elif args.mode == 2:

    result = args.num * 2

else:

    result = args.num

print(result)

运行结果：