argparse库中默认只支持字符串类型(string),但是我们可以通过添加type参数来指定不同的数据类型,比如int、float等。以下是一个示例代码,展示如何在argparse中支持float、int、string类型的输入。
import argparse
parser = argparse.ArgumentParser()
# 添加参数类型,支持float、int、string
parser.add_argument('-f', '--float', type=float, help='float type argument')
parser.add_argument('-i', '--int', type=int, help='int type argument')
parser.add_argument('-s', '--string', type=str, help='string type argument')
args = parser.parse_args()
# 读取命令行输入的参数
print("float argument = ", args.float)
print("int argument = ", args.int)
print("string argument = ", args.string)
在命令行运行该脚本,并添加参数,示例如下:
python test.py -f 3.14 -i 10 -s "Hello World"
输出结果为:
float argument = 3.14
int argument = 10
string argument = Hello World