### 使用 Python 的 argparse 模块解析命令行参数 为了实现命令行工具的功能，`argparse` 是一个非常强大的模块。下面是一个完整的例子来展示如何创建并使用 `argparse.ArgumentParser()` 来定义和解析命令行选项。 ```python import argparse def main(): # 创建 ArgumentParser 对象 parser = argparse.ArgumentParser(description='Process some integers.') # 添加位置参数 parser.add_argument('integers', metavar='N', type=int, nargs='+', help='an integer for the accumulator') # 添加可选参数 parser.add_argument('--sum', dest='accumulate', action='store_const', const=sum, default=max, help='sum the integers (default: find the max)') # 定义其他特定任务所需的参数 parser.add_argument("--data", default=None, required=True, type=str, help="数据文件夹"[^2]) parser.add_argument("--do_train", action='store_true', help="是否进行训练") parser.add_argument("--cuda_ids", default="0,1,2,3", type=str, help="指定 CUDA 设备 ID 列表") parser.add_argument("--epoch_num", default=20, type=int, help="设置训练轮数") parser.add_argument("--lr", default=1e-3, type=float, help="设定学习率") args = parser.parse_args() print(f'传入的数据路径为 {args.data}') if args.do_train: print("将执行训练过程.") else: print("不执行训练.") if __name__ == '__main__': main() ``` 这段代码展示了如何通过 `ArgumentParser` 类实例化对象，并调用其方法向该对象添加各种类型的命令行参数。这里既包含了必填的位置参数也涵盖了带有默认值的可选开关型参数以及具有具体含义的实际应用中的配置项。 当运行此脚本时，可以通过命令行传递相应的参数给它，比如： ```bash $ python script.py 1 2 3 --sum --data ./dataset/ --do_train --cuda_ids "0,1" ``` 这会触发上述条件下的输出逻辑，同时也会打印出所接收的所有参数及其对应的值。