深入解析 Python 的 argparse命令行参数的实战案例

import argparse
parser = argparse.ArgumentParser(description="这是一个演示脚本")
parser.add_argument("name", help="你的名字")
parser.add_argument("--age", type=int, help="你的年龄", default=18)
args = parser.parse_args()
print(f"你好, {args.name}！你今年 {args.age} 岁。")

python demo.py Alice --age 25

parser.add_argument("--count", type=int)

parser.add_argument("--city", default="Shanghai")

parser.add_argument("--verbose", action="store_true")

python demo.py --verbose

parser.add_argument("--mode", choices=["train", "test", "eval"])

parser.add_argument("--nums", nargs='+', type=int)

python demo.py --nums 1 2 3 4

parser.add_argument("filename")

python demo.py input.txt

parser.add_argument("-o", "--output")

python demo.py -o result.txt

python tool.py train --epochs 10
python tool.py test --model model.pth

parser = argparse.ArgumentParser()
subparsers = parser.add_subparsers(dest="command")
# 子命令 train
train_parser = subparsers.add_parser("train")
train_parser.add_argument("--epochs", type=int)
# 子命令 test
test_parser = subparsers.add_parser("test")
test_parser.add_argument("--model")
args = parser.parse_args()
if args.command == "train":
    print(f"开始训练，共 {args.epochs} 个 epoch")
elif args.command == "test":
    print(f"使用模型 {args.model} 进行测试")

group = parser.add_argument_group("训练参数")
group.add_argument("--batch_size", type=int)

parser.add_argument("--output", metavar="FILE", help="输出文件路径")

parser = argparse.ArgumentParser(
    description="示例",
    formatter_class=argparse.ArgumentDefaultsHelpFormatter
)

python demo.py --help

error: argument --age: invalid int value: 'abc'

try:
    args = parser.parse_args()
except SystemExit:
    print("请检查命令行参数是否正确")

import argparse
from PIL import Image
import os
parser = argparse.ArgumentParser(description="批量图像缩放工具")
parser.add_argument("input_dir", help="输入文件夹路径")
parser.add_argument("output_dir", help="输出文件夹路径")
parser.add_argument("--scale", type=float, default=0.5, help="缩放比例")
args = parser.parse_args()
if not os.path.exists(args.output_dir):
    os.makedirs(args.output_dir)
for file in os.listdir(args.input_dir):
    if file.lower().endswith((".png", ".jpg", ".jpeg")):
        img = Image.open(os.path.join(args.input_dir, file))
        new_size = (int(img.width * args.scale), int(img.height * args.scale))
        resized_img = img.resize(new_size)
        resized_img.save(os.path.join(args.output_dir, file))
print("处理完成！")

python resize.py ./input ./output --scale 0.8