在python中的字典列表中查找最小值

2022-02-24 00:00:00 python list return

问题描述

我要查找并返回字符串中id的最小值,例如:

find_min_id([{"nonid": "-222", "id": 0}, {"id": -101}])
-101
find_min_id([{’id’: 63, 'id': 42}])
42

到目前为止我有这个:

def find_min_id(list):
    return min(list)

但这意味着:

{'id': -101}

并且我只需要最低ID的值。


解决方案

使用minkey参数:

def find_min_id(l):
    return min(l, key=lambda d: d.get("id", float('inf')))["id"]

这实际上会查找最小ID,并且无需创建新列表即可执行此操作。

唯一的问题是,列表中的元素可能没有'id'键。为此,我不得不使用.get("id", float('inf'))。因此,如果没有id键,该函数将返回inf,这可能不是我们所希望的。当给定一个空列表时,min()所做的是抛出一个异常,所以如果我们传递给它的所有判据都没有'id'键,我们可能也会这样做。在这种情况下,生成器调用的最小值可能确实更好:

def find_min_id(l):
    return min(d["id"] for d in l if "id" in d)

另一种方法是检查min的结果是否为inf,但这比较麻烦:

import math
def find_min_id(l):
    res = min(l, key=lambda d: d.get("id", float('inf')))["id"]
    if math.isinf(res):
        raise ValueError("No dict has an 'id' key")
    return res

相关文章