Python fonction min avec une liste d'objets

Comment pouvez utiliser le key argument pour le min fonction pour comparer une liste d'objets de 1 attribut?

Exemple

class SpecialNumber:
    def __init__(self, i):
        self.number = i

li = [SpecialNumber(1), SpecialNumber(3), SpecialNumber(2)]
InformationsquelleAutor Pwnna | 2011-05-22
  1. 40

    http://docs.python.org/library/operator.html#operator.attrgetter

    from operator import attrgetter
min_num = min(li,key=attrgetter('number'))

    Exemple de session interactive:

    >>> li = [SpecialNumber(1), SpecialNumber(3), SpecialNumber(2)]
>>> [i.number for i in li]
[1, 3, 2]
>>> min_num = min(li,key=attrgetter('number'))
>>> print min_num.number
1
    • est le operator.attrgetter fonction plus rapide que d'utiliser une fonction lambda?
    InformationsquelleAutor bernie
  2. 33

    C'est:

    min(li, key=lambda x: x.number)

    vous avez besoin d'une fonction qui accepte un SpecialNumber et retourne son élément.

    • Est-il possible de ne pas retourner le nombre d'élément par retour de l' SpecialNumber classe?
    • Il ne retour au SpecialNumber objet par défaut. D'obtenir le plus petit élément, il suffit d'exécuter: min(li, key=lambda x: x.number).number
    InformationsquelleAutor viraptor
  3. 13

    Je le ferais en substituant __cmp__

    class SpecialNumber:
    def __init__(self, i):
        self.number = i

    def __repr__(self):
        return '<SpecialNumber(%d)>' % self.number

    def __cmp__(self, other):
        return cmp(self.number, other.number)

li = [SpecialNumber(1), SpecialNumber(3), SpecialNumber(2)]
print min(li) # <SpecialNumber(1)>
    • Notez que pour Python 3.0 cpm est allé. Ainsi, vous pouvez utiliser lt, eq et la functools.total_ordering décorateur sur la classe pour le faire fonctionner (ou de définir des lt, gt, le, ge, eq, ne)
    • ceux-ci devraient avoir été double souligne-apparemment, c'est markdown pour le gras...
    InformationsquelleAutor Daenyth
  4. 1

    la getattr version est plus rapide

    import random
from operator import attrgetter

class Test:
    def __init__(self):
        self.a = random.random()

t = [Test() for i in range(10000)]

%timeit min(t, key=lambda x: x.a)
1000 loops, best of 3: 790 µs per loop

%timeit min(t,key=attrgetter('a'))
1000 loops, best of 3: 582 µs per loop
    InformationsquelleAutor Davoud Taghawi-Nejad