如何在Python中为两个循环变量分配任意两个命令行参数值

核心要处理两个关键点：一是强制用户必须且只能传两个参数，二是自动把这两个参数的值分配给嵌套循环的两组变量，不用管具体是哪两个arg。

参数合法性校验 #

先把用户实际传入的非空参数收集起来，要是数量不是2，直接抛出错误提示，避免非法输入。

自动分配循环变量 #

把收集到的两个参数值直接解包给两组start/stop/num_samples，不用逐个判断arg1/arg2这些，代码更简洁灵活。

完整可运行代码 #

import numpy as np
import argparse

parser = argparse.ArgumentParser(description='Test')
parser.add_argument('-arg1', nargs=3, required=False, type=int)
parser.add_argument('-arg2', nargs=3, required=False, type=int)
parser.add_argument('-arg3', nargs=3, required=False, type=int)
parser.add_argument('-arg4', nargs=3, required=False, type=int)
args = parser.parse_args()

# 筛选出用户实际传入的参数值
provided_args = [val for val in vars(args).values() if val is not None]

# 校验参数数量必须是2
if len(provided_args) != 2:
    parser.error("必须且仅能传入两个参数，示例：python script.py -arg1 1 10 5 -arg4 2 20 10")

# 直接把两个参数的值分配给两组循环变量
(start1, stop1, num_samples1), (start2, stop2, num_samples2) = provided_args

# 执行嵌套循环
for a in np.linspace(start1, stop1, num_samples1):
    for b in np.linspace(start2, stop2, num_samples2):
        # 这里替换成你的业务逻辑，比如打印测试
        print(f"当前a值: {a:.2f}, 当前b值: {b:.2f}")

优势说明 #

• 适配任意两个参数的组合，不管用户传arg1+arg2还是arg3+arg4，都能自动适配，后续新增参数也不用修改逻辑。
• 用parser.error抛出的错误会自动带上argparse的帮助信息，用户更容易理解正确用法。
• 去掉了冗余的逐个参数判断，代码逻辑清晰，维护成本低。

内容的提问来源于stack exchange，提问作者phalanxasks