在Python中为字典实现argmin

时间:2015-05-07 04:11:36

标签: python function dictionary

之前可能已经提到过,但有没有人知道在Python中实现argmin的优雅方式?也就是说,给定字典D将整数映射到整数,我想找到密钥k,使D [k]最小化。

例如:

d = {1: 100, 2:200}
argmin(d) = 1

1 个答案:

答案 0 :(得分:1)

def argmin(d):
    if not d: return None
    min_val = min(d.values())
    return [k for k in d if d[k] == min_val][0]

d = {1: 50, 2:100, 3:11}
min_index = argmin(d)

修改

min有一个可选的关键参数,因此您可以使用此参数:

min_index = min(d, key=d.get)