Предположим, у меня есть 5 объектов: obj1 к obj5:Эффективно найти наибольшее значение среди объектов, а также определить объект
obj1.x = 2.7
obj2.x = 0.9
obj3.x = 3.8
obj4.x = 1.2
obj5.x = 0.4
Как я могу найти (эффективно) наибольшее значение x, и определить соответствующий obj? Вот ожидаемый ответ будет:
x = 3.8, it belongs to obj3
Кстати, в конкретном случае, у меня есть 500 объекты.
Если у вас есть список, и вы должны индекс максимума, вы можете комбинировать max и enumerate сделать эквивалент numpy.argmax в чистом Python:
l = [obj1, obj2, obj3, ...]
i = max(enumerate(l), key=lambda x: x[1].x)[0]
i будет индекс, содержащий максимальное значение , так что вы можете распечатать его как
print('x = {}, belongs to obj{}'.format(l[i].x, i + 1))
В примере вы дали, i == 2, который печатает x = 3.8, belongs to obj3.
Спасибо @Mad физик. Можно ли сортировать объекты, а не только найти максимальный? – aura
Вы имеете в виду использование 's = sorted (l, key = lambda x: x.x)' ?. В этом случае max будет просто 's [-1]'. Если 'l' является изменяемой последовательностью, такой как' list' (как в моем примере), вы также можете сделать 'l.sort (key = lambda x: x.x)', и в этом случае он будет отсортирован на месте. –
очень приятно. Благодарю. – aura
Лучше всего, чтобы поместить их в коллекции (т.е. list, tuple, ..) и использовать max с operator.attrgetter, чтобы захватить объект с максимальным значением x:
from operator import attrgetter
l = obj1, obj2, obj3, obj4, obj5
o = max(l, key=attrgetter('x'))
Значение o возвращается в настоящее время соответствует один с максимальным значением атрибута x, т.е. obj3:
o == obj3 # True
в идеале, вы не должны «идентификатор «передать» объект с помощью его имени, которое может легко измениться, и если у вас есть список, который не упорядочен по имени, вы получите неправильные результаты.
Вместо этого вы должны указать класс, который был создан из другого атрибута, скажем "name", и определить для него __str__/__repr__, который печатает имя и значение.
Ницца! Всегда приятно использовать 'attrgetter' :) – miradulo
Если вы хотите избежать импорта,' key = lambda x: x.x' отлично работает. –
Слишком плохо, что нет встроенного эквивалента функции «argmax» numpy. –
У вас есть они в списке или какой-то структуре данных? – miradulo
Я опубликовал новый ответ, основанный на предположении, что вы идентифицируете объекты по их индексу в последовательности, а не по имени уникальной переменной. –
@MadPhysicist Спасибо за вашу любезную помощь. Можно ли сортировать объекты, а не только найти максимальный. – aura