Python - argmin / argmax для функции-члена

Question

Я хотел бы знать, существует ли "питонический" способ использования математического аргумента argmin / argmax для функции-члена без использования библиотеки, такой как numpy.

У меня есть класс с функцией-членом внутри, которая возвращает целое число. Я создаю экземпляры нескольких объектов этого класса. Я хотел бы знать, какой объект имеет меньшее возвращаемое значение для этого метода.

Пожалуйста, найдите ниже мой исходный код. Часть, которую я хотел бы улучшить, это сразу после тега Code, который я хотел бы улучшить. Этот код работает очень хорошо, но я уверен, что есть лучший способ сделать то же самое.

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""argmin example"""


class People(object):
    """People class"""
    ret_ages = {"Half": 60, "AlmostFull": 65, "Full": 71}  # years

    def __init__(self, name, age, ret_mode):
        super(People, self).__init__()

        if ret_mode not in self.ret_ages.keys():
            raise KeyError(ret_mode + " not in " + str(self.ret_ages.keys()))

        self.name = name
        self.age = age
        self.ret_mode = ret_mode

    def get_remaining_years(self):
        """
                Return how many years People have still to work before earning
                <rate> retirement.
                <rate> could be "Half", "Middle" or "Full".
        """
        try:
            return self.ret_ages[self.ret_mode] - self.age
        except KeyError:
            raise KeyError("rate has to be in " + str(self.ret_ages.keys()))


def main():
    """Main function"""
    people_list = [
        People("Juliette", 35, "Full"),
        People("Coralie", 26, "Half"),
        People("Laura", 27, "AlmostFull")
    ]

    # Debugging print
    for people in people_list:
        print people.name, "has still to work",\
            people.get_remaining_years(), "years."
    print
    # End of debugging print

    ############################
    # Code I'd like to improve #
    ############################

    people_closer_to_ret = people_list[0]
    minimum_remainining_years = people_closer_to_ret.get_remaining_years()

    for people in people_list:
        if people.get_remaining_years() < minimum_remainining_years:
            people_closer_to_ret = people
            minimum_remainining_years = people.get_remaining_years()

            minimum_remainining_years = people.get_remaining_years()

    ###################################
    # End of code I'd like to improve #
    ###################################

    print people_closer_to_ret.name, "will be retired soon !"


if __name__ == '__main__':
    main()

Вот вывод этого скрипта:

Juliette has still to work 36 years.
Coralie has still to work 34 years.
Laura has still to work 38 years.

Coralie will be retired soon !

Answer 1

Хороший способ написать этот код более питоническим способом - использовать функцию min, которая на самом деле может использоваться как функция argmin благодаря ее ключу параметра.

Если мы заменим код между тегом Код, который я хотел бы улучшить, и Концом кода, который я хотел бы улучшить, путем:

people_closer_to_ret = min(people_list,
                           key=lambda people: people.get_remaining_years()),

это работает отлично. Ключевой аргумент полезен для указания функции min, какой критерий она должна минимизировать.

Итак, ваш полный код следующий:

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""argmin example"""


class People(object):
    """People class"""
    ret_ages = {"Half": 60, "AlmostFull": 65, "Full": 71}  # years

    def __init__(self, name, age, ret_mode):
        super(People, self).__init__()

        if ret_mode not in self.ret_ages.keys():
            raise KeyError(ret_mode + " not in " + str(self.ret_ages.keys()))

        self.name = name
        self.age = age
        self.ret_mode = ret_mode

    def get_remaining_years(self):
        """
                Return how many years People have still to work before earning
                <rate> retirement.
                <rate> could be "Half", "Middle" or "Full".
        """
        try:
            return self.ret_ages[self.ret_mode] - self.age
        except KeyError:
            raise KeyError("rate has to be in " + str(self.ret_ages.keys()))


def main():
    """Main function"""
    people_list = [
        People("Juliette", 35, "Full"),
        People("Coralie", 26, "Half"),
        People("Laura", 27, "AlmostFull")
    ]

    # Debugging print
    for people in people_list:
        print people.name, "has still to work",\
            people.get_remaining_years(), "years."
    print
    # End of debugging print

    people_closer_to_ret = min(people_list,
                               key=lambda people: people.get_remaining_years())

    print people_closer_to_ret.name, "will be retired soon !"


if __name__ == '__main__':
    main()