python 定义参数的类型
时间: 2023-07-04 09:21:16 浏览: 111
在 Python 中,可以使用 argparse 模块来定义命令行参数的类型。argparse 模块支持的参数类型包括字符串、整数、浮点数、布尔值等。
下面是一些常用的参数类型及其使用方法:
- 字符串类型:使用 `type=str` 来指定参数类型为字符串,这是默认的参数类型,可以不用指定。
``` python
parser.add_argument('--name', type=str, help='Name')
```
- 整数类型:使用 `type=int` 来指定参数类型为整数。
``` python
parser.add_argument('--age', type=int, help='Age')
```
- 浮点数类型:使用 `type=float` 来指定参数类型为浮点数。
``` python
parser.add_argument('--score', type=float, help='Score')
```
- 布尔类型:使用 `action='store_true'` 来指定参数类型为布尔值,当命令行中存在该参数时,其值为 True,否则为 False。
``` python
parser.add_argument('--flag', action='store_true', help='Flag')
```
- 列表类型:使用 `nargs='+'` 来指定参数类型为列表,可以接受多个值。
``` python
parser.add_argument('--items', nargs='+', help='Items')
```
- 自定义类型:可以通过自定义函数来指定参数类型,该函数接受一个字符串参数,返回一个转换后的值。
``` python
def my_type(string):
# do something to convert string to desired type
return converted_value
parser.add_argument('--myarg', type=my_type, help='My Arg')
```
使用 argparse 模块可以很方便地定义和解析命令行参数,并且支持多种参数类型,可以满足大部分需求。
阅读全文