python - 是否可以将 numpy.argmax 与列表中对象的自定义字段一起使用？

Question

类似于:

class Test:
    def __init__(self, n):
        self.id = n

    def __str__(self):
        return str(self.id)

my_list = []
my_list.append(Test(1))
my_list.append(Test(2))
my_list.append(Test(3))

是否可以获取列表中id最大或最小的元素？

Answer 1

在您的示例中，您甚至不需要求助于 numpy.argmax()。由于您的对象在标准 Python 列表中，您还可以使用 Python 的内置 max() 函数:

index = max(range(len(my_list)), key=lambda i: my_list[i].id)

或

index = max(enumerate(my_list), key=lambda x: x[1].id)[0]

还将为您提供具有最大 id 的项目的索引。