如何实现可接收指定键参数并返回对应值的字典组装函数？

首先看原函数的实现逻辑：

def data_fetch():
    # The function assembles this dictionary from various sources:
    my_dict = {'a': [1, 2, 3],
               'b': [4, 5, 6],
               'c': [7, 8, 9]}

需求是：调用该函数时可传入a、b、c的任意组合参数，返回对应键的值（单个值直接返回，多个值返回元组）。下面是几种实用的实现方式：

方式一：可变位置参数（灵活通用） #

通过*args接收任意数量的键名参数，逻辑简单且扩展性强：

def data_fetch(*args):
    # 原有的字典组装逻辑保持不变
    my_dict = {'a': [1, 2, 3],
               'b': [4, 5, 6],
               'c': [7, 8, 9]}
    
    # 未传参数时可返回整个字典（按需调整，也可抛出异常）
    if not args:
        return my_dict
    
    # 收集指定键对应的值
    results = tuple(my_dict[key] for key in args)
    
    # 单个值返回元素本身，多个值返回元组
    return results[0] if len(results) == 1 else results

调用示例：

• data_fetch('a') → 返回 [1, 2, 3]
• data_fetch('a', 'b') → 返回 ([1, 2, 3], [4, 5, 6])
• data_fetch('c') → 返回 [7, 8, 9]

这种写法符合Python惯用风格，后续新增键无需修改函数参数，还能自然处理非法键（直接抛出KeyError，符合Python错误处理的直观性）。

方式二：关键字参数（语义清晰） #

如果希望调用时更直观，避免传错键名，可以用显式的关键字参数：

def data_fetch(a=False, b=False, c=False):
    my_dict = {'a': [1, 2, 3],
               'b': [4, 5, 6],
               'c': [7, 8, 9]}
    
    # 筛选出需要返回的键
    requested_keys = [key for key, flag in zip(['a', 'b', 'c'], [a, b, c]) if flag]
    
    if not requested_keys:
        return my_dict  # 或按需抛出异常
    
    results = tuple(my_dict[key] for key in requested_keys)
    
    return results[0] if len(results) == 1 else results

调用示例：

• data_fetch(a=True) → 返回 [1, 2, 3]
• data_fetch(a=True, c=True) → 返回 ([1, 2, 3], [7, 8, 9])

这种方式调用时语义明确，但缺点是后续新增键需要修改函数参数列表，扩展性稍弱。

方案选择建议 #

• 若键的数量固定且极少，两种方案都可行，关键字参数的可读性更优；
• 若后续可能新增键，或需要支持任意合法键，可变位置参数方案更灵活、规范，扩展性更强；
• 性能上两种方案差异可忽略，瓶颈仅在于原函数中组装my_dict的逻辑。

额外优化：可以添加键合法性检查，比如对不存在的键返回默认值或抛出明确的提示，但默认让KeyError抛出更符合Python的“快速失败”原则。

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