argparse

Argparse学习笔记

记一下一些常用的Argparse方法

1. 引入argparse

#test.py
import argparse#引入包
parser = argparse.ArgumentParser()#创建对象
parser.parse_args()#接收参数

python3 test.py --help
usage: test.py [-h]

optional arguments:
  -h, --help  show this help message and exit

2.加入一个参数

import argparse
parser = argparse.ArgumentParser()
parser.add_argument("a1", help="input a1")
args=parser.parse_args()
print(args.a1)

python3 test.py A1
A1

3.声明参数类型

import argparse
parser = argparse.ArgumentParser()
parser.add_argument("a1", help="input a1(int)",type=int)
args=parser.parse_args()
print(args.a1**2)

python3 test.py 7
49

4.声明可选参数

import argparse
parser = argparse.ArgumentParser()
parser.add_argument("--a1", help="input a1")
args = parser.parse_args()
if args.a1:
    print(args.a1)

python test.py --a1 A1
A1

5.短选项

import argparse
parser = argparse.ArgumentParser()
parser.add_argument("-a","--a1", help="input a1")
args = parser.parse_args()
if args.a1:
    print(args.a1)

python test.py -a A1
A1

6.限定值

import argparse
parser = argparse.ArgumentParser()
parser.add_argument("-a","--a1", help="input a1",choices=['A1','B1'])
args = parser.parse_args()
if args.a1:
    print(args.a1)

python test.py -a C1
usage: test.py [-h] [-a {A1,B1}]
test.py: error: argument -a/--a1: invalid choice: 'C1' (choose from 'A1', 'B1')