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] Python Tutorial - Argparse