如何在Python中使用key获取最小值及其对应key值（避免重复计算）

可以通过预计算每个元素的key值并配对存储的方式，避免重复调用高成本的key函数。具体思路是：将集合中的每个元素与它的key值组成元组（key_val, element），然后对这些元组取最小值——因为Python中元组比较会优先对比第一个元素（也就是我们的排序key），这样就能一次性拿到最小key对应的元素和key值，无需二次计算。

示例代码 #

# 目标集合
target_set = {"0", "00", "000"}
# 高成本的key函数（这里用len示例，实际替换为你的高成本函数）
key_func = len

# 生成包含key值和元素的元组，取最小值
min_pair = min((key_func(item), item) for item in target_set)
# 调整顺序得到需要的 (元素, key值) 结果
result = (min_pair[1], min_pair[0])

print(result)  # 输出: ('0', 1)

说明 #

• 这种方式只对每个元素调用一次key函数，彻底避免了二次计算的开销。
• 如果多个元素的key值相同，min会返回迭代过程中第一个遇到的元素，这和Python原生min()函数的行为保持一致。

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