python命令行参数解析工具——argparse

在命令行中运行py文件的时候(这在linux系统中尤为常见),是否可以像运行内部的函数那样设置可变参数呢?这样做可以避免进入py文件再次编写文件的尴尬,同时也意味着减少bug的发生。

argparse就是这样可以在帮助在命令行中添加参数的工具,基本的使用方法如下代码所示:

# -*- coding: utf-8 -*- 
import math
import argparse

# argparse.ArgumentParser创建一个参数容器
parser = argparse.ArgumentParser(description= parser是存放参数的容器)

# add_argument增加可选参数,
parser.add_argument(radius, type=int, help=圆柱体半径)
parser.add_argument(height, type=int, help=圆柱体高度)

# 解析参数
args = parser.parse_args()

def cylinder_vol(radius, height):
    """计算圆柱体体积"""
    vol = (math.pi) * (radius ** 2) * (height)
    return vol

if __name__== __main__:
    print (cylinder_vol(args.radius, args.height))

之后我们要做的,就是在运行程序时加上参数(半径2,高度3):

运行结果如下:

当然,有时候由于参数太多,我们可能会记错参数的位置,那么给定参数名再赋值会是个更好的选择。我们只需要在add_argument的参数名之前加入--即可,比如 radius 改为 --radius:

parser.add_argument(--radius, type=int, help=圆柱体半径)
parser.add_argument(--height, type=int, help=圆柱体高度)

运行代码命令:

python test.py --radius=2 --height=2

结果为:

25.132741228718345

关于argparse的其它一些用法,诸如参数命名缩写等等,可以查阅参考资料中的讲解,这里就不多说了。

参考资料:

[1]

经验分享 程序员 微信小程序 职场和发展