是否可以使用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索引。

Answer 2

如果您爲類Test定義__cmp__運算符，則argmax應該按預期運行。