python如何从命令行输入参数

Python从命令行输入参数的方法包括使用sys.argv、argparse模块、以及click模块等。其中，最常用的是sys.argv和argparse模块。sys.argv适用于简单的参数传递，argparse模块则用于处理更复杂的参数解析需求。接下来，我们将详细介绍这些方法，并通过具体的代码示例进行说明。

一、sys.argv

sys.argv是Python标准库中的一个列表，用于存储命令行参数。第一个元素是脚本的名称，从第二个元素开始是传递给脚本的参数。下面是使用sys.argv的示例：

import sys
def main():
    if len(sys.argv) < 2:
        print("Usage: python script.py <arg1> <arg2> ...")
        sys.exit(1)
    print("Script name:", sys.argv[0])
    for i, arg in enumerate(sys.argv[1:], start=1):
        print(f"Argument {i}:", arg)
if __name__ == "__main__":
    main()

在命令行中运行该脚本并传递参数：

python script.py param1 param2 param3

输出将会是：

Script name: script.py Argument 1: param1 Argument 2: param2 Argument 3: param3

二、argparse模块

argparse模块是Python内置的一个用于解析命令行参数的模块。它比sys.argv功能更强大，支持参数类型、默认值、帮助信息等。

1、基本使用

以下是一个简单的示例，展示如何使用argparse解析命令行参数：

import argparse
def main():
    parser = argparse.ArgumentParser(description="A simple argparse example")
    parser.add_argument("name", type=str, help="Your name")
    parser.add_argument("age", type=int, help="Your age")
    args = parser.parse_args()
    print(f"Name: {args.name}")
    print(f"Age: {args.age}")
if __name__ == "__main__":
    main()

在命令行中运行该脚本并传递参数：

python script.py John 30

输出将会是：

Name: John Age: 30

2、可选参数

argparse还支持可选参数，可以使用--或-前缀来定义。以下是一个示例：

import argparse
def main():
    parser = argparse.ArgumentParser(description="A simple argparse example with optional arguments")
    parser.add_argument("--name", type=str, help="Your name", default="Unknown")
    parser.add_argument("--age", type=int, help="Your age", default=0)
    args = parser.parse_args()
    print(f"Name: {args.name}")
    print(f"Age: {args.age}")
if __name__ == "__main__":
    main()

在命令行中运行该脚本并传递参数：

python script.py --name John --age 30

输出将会是：

Name: John Age: 30

三、click模块

click模块是一个第三方库，用于创建命令行界面（CLI）。它比argparse更易用，功能更强大。首先需要安装click库：

pip install click

1、基本使用

以下是一个简单的示例，展示如何使用click解析命令行参数：

import click
@click.command()
@click.argument('name')
@click.argument('age', type=int)
def main(name, age):
    click.echo(f"Name: {name}")
    click.echo(f"Age: {age}")
if __name__ == "__main__":
    main()

在命令行中运行该脚本并传递参数：

python script.py John 30

输出将会是：

Name: John Age: 30

2、可选参数

click还支持可选参数，可以使用--或-前缀来定义。以下是一个示例：

import click
@click.command()
@click.option('--name', default='Unknown', help='Your name')
@click.option('--age', default=0, help='Your age', type=int)
def main(name, age):
    click.echo(f"Name: {name}")
    click.echo(f"Age: {age}")
if __name__ == "__main__":
    main()

在命令行中运行该脚本并传递参数：

python script.py --name John --age 30

输出将会是：

Name: John Age: 30

四、参数验证和默认值

无论是使用sys.argv、argparse还是click，都可以为参数设置默认值和进行验证。例如，使用argparse时，可以为参数设置默认值，并对参数类型进行验证：

import argparse
def main():
    parser = argparse.ArgumentParser(description="A simple argparse example with default values and validation")
    parser.add_argument("--name", type=str, help="Your name", default="Unknown")
    parser.add_argument("--age", type=int, help="Your age", default=0, choices=range(1, 100))
    args = parser.parse_args()
    print(f"Name: {args.name}")
    print(f"Age: {args.age}")
if __name__ == "__main__":
    main()

在命令行中运行该脚本并传递参数：

python script.py --name John --age 30

输出将会是：

Name: John Age: 30

如果传递的年龄不在1到99之间，会输出错误信息：

python script.py --name John --age 150

输出将会是：

usage: script.py [-h] [--name NAME] [--age {1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31,32,33,34,35,36,37,38,39,40,41,42,43,44,45,46,47,48,49,50,51,52,53,54,55,56,57,58,59,60,6