包含min函数的栈-python

#思路：新建一个辅助栈来存储当前最小值，当有更小值入栈时，将其压入此栈，最小值为栈顶元素。当栈顶的元素出栈后，此时栈顶的元素就顺应成为新的最小值
class min_stack:
    def __init__(self):
        self.stack = []
        self.minstack = []

    def push(self, data):
        # 将新元素添加到栈
        self.stack.append(data)
        # 当新元素比之前的元素小时，将新元素压入到辅助栈，否则将之前元素重复压入辅助栈
        if data < self.minstack[-1] or len(self.minstack) == 0:
            self.minstack.append(data)
        else:
        	self.minstack.append(self.minstack[-1])

    def pop(self):
        self.minstack.pop()
        return self.stack.pop()
    
    def min(self):
        return self.minstack[-1]