Emplacements des objets avec des classes (Python)

Question

Lorsque j'imprime ma liste des vingt occurrences les plus importantes, tout ce que j'obtiens est l'emplacement de l'objet et non les données réelles dont j'ai besoin. Aussi, si vous avez des idées d'amélioration, cela serait grandement apprécié. S'il vous plaît aider. Merci.

Voici mon code.

class topList(): 
    __slots__ = ("name", "gender", "occurences") 

def mkList(name, gender, occurences): 
    find = topList() 
    find.name = name 
    find.gender = gender 
    find.occurences = occurences 
    return find 

def main(): 
    year = input('Enter year: ') 
    file = open('yob' + year + '.txt') 
    lst = [] 
    femaleLst = [] 
    maleLst = [] 
    for line in file: 
     line = line.strip().split(",") 
     names = mkList(line[0], line[1], line[2]) 
     lst.append(names) 
     if names.gender == 'F': 
      femaleLst += [ line ] 
     else: 
      maleLst += [ line ] 
    while len(lst) < 20: 
     male = maleLst.pop() 
     female = femaleLst.pop() 
     if maleLst.occurences > femaleLst.occurences: 
      lst += [ male ] 
     else: 
      lst += [ female ] 
    print(lst[ : 20]) 

main() 

Answer 1

Ajouter __repr__ et __str__ méthodes à votre classe:

class topList(): 
    __slots__ = ("name", "gender", "occurences") 

    def __repr__(self): 
     return 'topList({s.name!r}, {s.gender!r}, {s.occurences!r})'.format(s=self) 

    def __str__(self): 
     return '{s.name} ({s.gender}): {s.occurences!r})'.format(s=self) 

Démo:

>>> class topList(): 
...  __slots__ = ("name", "gender", "occurences") 
...  def __str__(self): 
...   return 'topList({s.name!r}, {s.gender!r}, {s.occurences!r})'.format(s=self) 
... 
>>> tl = topList() 
>>> tl.name = 'FooBar' 
>>> tl.gender = 'm' 
>>> tl.occurences = 42 
>>> print(tl) 
FooBar (m): 42 
>>> tl 
topList('FooBar', 'm', 42) 

La méthode __str__ est appelée automatiquement lorsque votre objet est converti en une chaîne, comme le print() la fonction fait.

La méthode __repr__ est utilisée pour les représentations repr() et lorsque votre objet est affiché dans des listes et autres conteneurs.