之前可能已经提到过,但有没有人知道在Python中实现argmin的优雅方式?也就是说,给定字典D将整数映射到整数,我想找到密钥k,使D [k]最小化。
例如:
d = {1: 100, 2:200}
argmin(d) = 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)