스택 구현하기

class Stack(list):
    push = list.append
 
    def is_empty(self):
        if not self:
            return True
        else:
            return False
 
    def peek(self):
        return self[-1]
 
if __name__ == "__main__":
    s = Stack()
    s.push(1)
    s.push(2)
    s.push(3)
 
    while s:
        data = s.pop()
        print(data, end='')
 

리스트를 상속받아 스택을 구현한 것이다.

스택 클래스를 한 번 구현해놓으면 import해서 사용하기 편리하다.

 

if __name__ == "__main__" 이 부분은

이 소스코드가 있는 파이썬 파일이 다른 파일로부터 호출된것이 아닌 메인으로 실행됐을때 동작하게 하기 위해 쓰는 것으로, 스택이 잘 동작하는지 테스트하기 위해 사용했다.