如何用字典模拟Python的switch-case？求修复并简化参数校验代码

首先，我们来拆解你代码里的几个核心问题，然后一步步优化到符合预期的状态：

你的代码为什么无法运行？ #

1. 字典Key设计完全错误
   你用len(args) == 1这类布尔表达式作为字典Key，但布尔值只有True和False两种。这意味着多个case会共享同一个Key（比如当参数长度为2时，len(args)==2是True，会覆盖之前len(args)==1对应的True Key），字典最终只会保留最后一个True对应的逻辑，完全无法区分不同的参数长度场景。

2. get方法的参数不匹配
   你调用.get(args, ...)，但args是传入的参数元组，和字典的布尔Key完全不匹配，所以永远只会触发默认的lambda分支，这是代码失效的核心原因。

3. 返回值与解包逻辑不兼容
   比如len(args)==2对应的lambda返回(args, 1)，这是一个二元组，但你要解包给三个变量start, stop, step，会直接抛出ValueError: not enough values to unpack。

4. 异常抛出逻辑错误
   默认分支的lambda: TypeError("xxx...")只是创建了一个TypeError对象，并没有主动抛出异常，无法达到报错的目的。

修正并简化后的版本 #

我们用参数长度数值作为字典Key（这才是模拟switch-case的正确思路，类似匹配不同的case值），同时修正逻辑细节：

版本1：最直观的实现（推荐）

def _validate_inputs(*args):
    n = len(args)
    try:
        # 用参数长度作为Key，直接映射到对应的返回元组
        start, stop, step = {
            1: (1, args[0], 1),    # 对应原逻辑：start=1, stop=单个参数值, step=1
            2: (*args, 1),         # 对应原逻辑：start/stop取前两个参数，step=1
            3: args                # 对应原逻辑：直接使用三个参数
        }[n]
    except KeyError:
        # 参数长度不在1-3范围内时，抛出指定TypeError
        raise TypeError(f"xxx expected at most 3 arguments, got {n}")
    
    if step == 0:
        raise ValueError("xxx arg 3 must not be zero")
    return start, stop, step

版本2：用get方法合并异常逻辑（更贴近switch-case写法）

如果想把异常处理也塞进字典逻辑里，可以用lambda配合生成器抛出异常（兼容Python全版本）：

def _validate_inputs(*args):
    n = len(args)
    # get方法找不到Key时，执行lambda抛出异常
    start, stop, step = {
        1: (1, args[0], 1),
        2: (*args, 1),
        3: args
    }.get(n, lambda: (_ for _ in ()).throw(TypeError(f"xxx expected at most 3 arguments, got {n}")))()
    
    if step == 0:
        raise ValueError("xxx arg 3 must not be zero")
    return start, stop, step

补充说明 #

• 我把原逻辑中len(args)==1时的stop=args（元组）改成了stop=args[0]（单个值），这更符合range函数的参数逻辑，如果你需要严格和原_validate_inputs2保持一致，改回(1, args, 1)即可。
• 这种字典模拟switch-case的写法，核心是用唯一可枚举的Key（比如数值、字符串）对应不同分支逻辑，避免布尔Key的冲突问题。

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