python脚本如何添加参数运行程序

Python脚本添加参数运行程序的方法包括：使用sys.argv、argparse模块、使用click库。 其中，使用argparse模块是最常见和推荐的方法，因为它提供了强大的功能来解析命令行参数，并且可以自动生成帮助和使用信息。下面将详细介绍如何使用argparse模块来添加参数并运行Python脚本。

一、使用 `sys.argv`

sys.argv 是一个列表，其中包含命令行参数。第一个元素是脚本名称，后续元素是传递给脚本的参数。

import sys
def main():
    if len(sys.argv) != 3:
        print("Usage: python script.py <param1> <param2>")
        sys.exit(1)
    param1 = sys.argv[1]
    param2 = sys.argv[2]
    print(f"Parameter 1: {param1}")
    print(f"Parameter 2: {param2}")
if __name__ == "__main__":
    main()

在命令行运行脚本：

python script.py arg1 arg2

二、使用 `argparse` 模块

argparse 模块是 Python 标准库中用于解析命令行参数的模块。它提供了更加灵活和强大的方式来处理命令行参数。

1. 基本使用

import argparse
def main():
    parser = argparse.ArgumentParser(description="A simple argument parser")
    parser.add_argument('param1', type=str, help='First parameter')
    parser.add_argument('param2', type=str, help='Second parameter')
    args = parser.parse_args()
    print(f"Parameter 1: {args.param1}")
    print(f"Parameter 2: {args.param2}")
if __name__ == "__main__":
    main()

在命令行运行脚本：

python script.py arg1 arg2

2. 添加可选参数

除了位置参数，argparse 还可以处理可选参数。

import argparse
def main():
    parser = argparse.ArgumentParser(description="A simple argument parser")
    parser.add_argument('--param1', type=str, required=True, help='First parameter')
    parser.add_argument('--param2', type=str, required=True, help='Second parameter')
    args = parser.parse_args()
    print(f"Parameter 1: {args.param1}")
    print(f"Parameter 2: {args.param2}")
if __name__ == "__main__":
    main()

在命令行运行脚本：

python script.py --param1 arg1 --param2 arg2

3. 默认值和类型转换

argparse 还支持为参数设置默认值和进行类型转换。

import argparse
def main():
    parser = argparse.ArgumentParser(description="A simple argument parser")
    parser.add_argument('--param1', type=int, default=10, help='First parameter (default: 10)')
    parser.add_argument('--param2', type=float, default=20.5, help='Second parameter (default: 20.5)')
    args = parser.parse_args()
    print(f"Parameter 1: {args.param1}")
    print(f"Parameter 2: {args.param2}")
if __name__ == "__main__":
    main()

在命令行运行脚本：

python script.py --param1 15 --param2 30.5

如果不提供参数，将使用默认值：

python script.py Output: Parameter 1: 10 Parameter 2: 20.5

三、使用 `click` 库

click 是一个第三方库，用于创建美观的命令行界面。它比 argparse 更加简洁和易用。

1. 安装 `click`

在使用 click 之前，需要先安装它：

pip install click

2. 使用 `click` 创建命令行接口

import click
@click.command()
@click.option('--param1', default='default1', help='First parameter')
@click.option('--param2', default='default2', help='Second parameter')
def main(param1, param2):
    print(f"Parameter 1: {param1}")
    print(f"Parameter 2: {param2}")
if __name__ == "__main__":
    main()

在命令行运行脚本：

python script.py --param1 arg1 --param2 arg2

如果不提供参数，将使用默认值：

python script.py Output: Parameter 1: default1 Parameter 2: default2