参考代码:
# 实现栈
class Stack(object):
""" 栈 """
def __init__(self):
# 防止用户直接对栈容器进行操作,故用私有变量
self.__list = []
def push(self, item):
"""添加一个新的元素item到栈顶"""
self.__list.append(item)
def pop(self):
"""弹出栈顶元素"""
return self.__list.pop()
def peek(self):
""" 返回栈顶元素"""
# 如果列表不为空
if self.__list:
return self.__list[-1]
else:
return None
def is_empty(self):
""" 判断栈顶是否为空 """
return self.__list == []
# return not self.__list # 等效于上面的语句
def size(self):
"""返回栈的元素个数"""
return len(self.__list)
if __name__ == "__main__":
s = Stack()
# 压栈
s.push(1)
s.push(2)
s.push(3)
s.push(4)
# 出栈
print(s.pop())
print(s.pop())
print(s.pop())
# 取栈顶元素
print(s.peek())
# 再出栈
print(s.pop())
# 再取栈顶元素
print(s.peek())
ps:栈就是一种对数据的操作方式。