[python] class instance의 변수로 정렬하기

게임 캐릭터를 만든다 생각하고 캐릭터를 만들어보자

class charactor(object):
    def __init__(self,id,name,hp,mp,power):
        self.id = id
        self.name = name
        self.hp = hp
        self.mp = mp
        self.power = power
 
a = charactor(id=1, name='minho', hp=300, mp=300, power=10)
b = charactor(id=2, name='bonggu', hp=300, mp=200, power=80)

class charactor(object):
    def __init__(self,id,name,hp,mp,power):
        self.id = id
        self.name = name
        self.hp = hp
        self.mp = mp
        self.power = power
    def __lt__(self,other):
        return self.power > other.power
 
class stronger(object):
    def selectStronger(self,charactors):
        sortedCharactors = sorted(charactors)
        for i,v in enumerate(sortedCharactors):
            print(i,v.name,v.power)
        
 
a = charactor(id=1, name='minho', hp=300, mp=300, power=10)
b = charactor(id=2, name='bonggu', hp=300, mp=200, power=80)
c = stronger()
c.selectStronger([a,b])
 

캐릭터 클래스에 __lt__ 라는 메소드를 오버라이드 해서 비교할 값을 넣어주면 된다.

또는 sorted와 lambda를 활용하여 이렇게 할 수 있다.

 

 

class charactor(object):
    def __init__(self,id,name,hp,mp,power):
        self.id = id
        self.name = name
        self.hp = hp
        self.mp = mp
        self.power = power
 
class stronger(object):
    def selectStronger(self,charactors):
        sortedCharactors = sorted(charactors,key=lambda x : x.power,reverse=True)
        for i,v in enumerate(sortedCharactors):
            print(i,v.name,v.power)
        
 
a = charactor(id=1, name='minho', hp=300, mp=300, power=10)
b = charactor(id=2, name='bonggu', hp=300, mp=200, power=80)
c = stronger()
c.selectStronger([a,b])
 

 

또는 collector 모듈의 attrgetter 메소드를 활용할 수도 있다.

 

from operator import attrgetter
 
class charactor(object):
    def __init__(self,id,name,hp,mp,power):
        self.id = id
        self.name = name
        self.hp = hp
        self.mp = mp
        self.power = power
 
class stronger(object):
    def selectStronger(self,charactors):
        sortedCharactors = sorted(charactors,key=attrgetter('power'),reverse=True)
        for i,v in enumerate(sortedCharactors):
            print(i,v.name,v.power)
        
 
a = charactor(id=1, name='minho', hp=300, mp=300, power=10)
b = charactor(id=2, name='bonggu', hp=300, mp=200, power=80)
c = stronger()
c.selectStronger([a,b])