Python：如何交换枚举展开的变量位置以统一遍历集合与字典？

问题场景 #

需要处理两种类型的对象模拟C/C++枚举：

• Dict类型示例：{'a': 5, 'b': 2, 'c': 8}，遍历需取key, value（对应枚举的名称和值）
• Set类型示例：{'a', 'b', 'c'}，遍历需取元素, 索引（索引作为枚举值，元素作为名称）

原实现需写两个仅变量顺序不同的循环，希望通过统一迭代器简化代码，但enumerate(set)返回的是(索引, 元素)，和dict.items()的(key, value)顺序不匹配。

解决方案 #

方法1：生成器表达式交换enumerate的元素顺序 #

利用生成器将enumerate返回的元组反转，让迭代器输出格式和dict.items()一致：

myenum = {'a', 'b', 'c'}  # 或 {'a':5, 'b':2, 'c':8}
itr = myenum.items() if isinstance(myenum, dict) else ((k, i) for i, k in enumerate(myenum))

for k, i in itr:
    print(f"枚举名称: {k}, 枚举值: {i}")

方法2：使用map+lambda转换元组顺序 #

也可以用map配合匿名函数反转enumerate的元组：

myenum = {'a', 'b', 'c'}  # 或 {'a':5, 'b':2, 'c':8}
itr = myenum.items() if isinstance(myenum, dict) else map(lambda x: (x[1], x[0]), enumerate(myenum))

for k, i in itr:
    print(f"枚举名称: {k}, 枚举值: {i}")

说明 #

• 用isinstance(myenum, dict)替代hasattr(myenum, 'keys')更严谨，避免误判其他含keys方法的对象
• 两种方法都能让迭代器统一输出(名称, 值)的格式，实现单循环处理两种输入类型

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