使用itertools.product仅传单个集合时，如何解包生成的单元素元组以设置字典默认值？

针对itertools.product单集合输出单元素元组导致的键类型不匹配问题，不需要针对具体字典、集合做单独适配，用以下两种通用逻辑即可覆盖所有场景：

方案1：循环内加统一键转换（改动最小，无需修改原有调用逻辑） #

完全保留现有itertools.product的调用方式，仅在循环拿到遍历结果后，增加一行通用的键类型判断逻辑：

• 对遍历返回的元组，若长度为1，直接取内部唯一值作为实际键值，适配单值键场景
• 若元组长度大于1，保留原元组作为实际键值，适配多值元组键场景
  代码示例：

import itertools

# 原有问题场景的修正实现
set_a = {1,2,3,4}
dict_y = {1:'blah',2:'blahblah'}

for product_key in itertools.product(set_a):
    # 通用键转换逻辑
    key = product_key[0] if len(product_key) == 1 else product_key
    dict_y.setdefault(key, 'hi')

print(dict_y)
# 最终输出: {1:'blah', 2:'blahblah', 3:'hi', 4:'hi'}

场景兼容性说明：

• 多集合遍历、元组键字典的场景：遍历返回的元组长度大于1，会直接作为键传入setdefault，和原有逻辑完全一致，无兼容性问题
• 单集合遍历、单值键字典的场景：自动解包单元素元组，不会错误生成(1,)这类冗余键

注意：不要直接对任意长度的元组做通用解包，避免多元素元组的键被打散，导致和原有元组键的匹配逻辑冲突。

方案2：封装通用product调用入口（从根源避免单元素元组生成） #

如果可以调整itertools.product的调用入口，可封装一层通用判断逻辑，在仅传入单个集合时直接返回集合本身的元素，不做元组封装：

def generic_product(*iterables):
    # 仅传入单个可迭代对象时，直接返回元素本身
    if len(iterables) == 1:
        return iterables[0]
    # 多可迭代对象时，走原生product逻辑
    return itertools.product(*iterables)

使用时把原来的itertools.product(xxx)替换成generic_product(xxx)即可：

for key in generic_product(set_a):
    dict_y.setdefault(key, 'hi')

效果说明：

• 单集合调用场景：直接返回原始单值作为遍历结果，从根源上避免单元素元组的生成
• 多集合调用场景：返回结果和原生itertools.product完全一致，不会影响原有元组键的匹配逻辑

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