跳到主要内容

Python处理命令行参数示例代码?

参考答案:

在Python中,你可以使用内置的argparse模块来处理命令行参数。下面是一个基本的示例代码:

import argparse

def main():
    # 创建解析器对象
    parser = argparse.ArgumentParser(description='这是一个示例程序,用于演示如何处理命令行参数')

    # 添加参数
    parser.add_argument('--input', type=str, help='输入文件的路径')
    parser.add_argument('--output', type=str, help='输出文件的路径')
    parser.add_argument('--verbose', action='store_true', help='显示更多信息')

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

    # 处理参数
    if args.verbose:
        print('Verbose mode is on.')

    if args.input is not None and args.output is not None:
        print(f'Reading from {args.input} and writing to {args.output}')
    else:
        print('Please provide both input and output file paths.')

if __name__ == '__main__':
    main()

这个示例代码定义了一个简单的命令行程序,它接受两个位置参数(--input--output)以及一个可选的布尔参数(--verbose)。在main函数中,我们首先创建了一个ArgumentParser对象,并使用add_argument方法添加了需要的参数。然后,我们使用parse_args方法解析命令行参数,并将解析结果存储在args对象中。最后,我们根据args对象中的参数值进行相应的处理。

要运行这个程序,你可以在命令行中输入以下命令:

python your_script.py --input input.txt --output output.txt --verbose

这将输出以下结果:

Verbose mode is on.
Reading from input.txt and writing to output.txt

如果你省略了某个参数,比如--output,程序将输出一个错误消息,提示你提供所有必要的参数。