python中stack的方法
时间: 2023-07-14 09:08:33 浏览: 235
Python中可以使用列表来实现栈的功能,下面是一些常用的栈方法:
1. push(element): 将元素压入栈顶,即向列表末尾添加元素。
2. pop(): 弹出栈顶元素,并返回该元素。如果栈为空,则会引发 IndexError 异常。
3. peek(): 返回栈顶元素,但不删除它。如果栈为空,则会引发 IndexError 异常。
4. is_empty(): 检查栈是否为空,如果栈为空则返回 True,否则返回 False。
5. size(): 返回栈中元素的数量。
下面是一个示例代码:
```python
class Stack:
def __init__(self):
self.stack = []
def push(self, element):
self.stack.append(element)
def pop(self):
if not self.is_empty():
return self.stack.pop()
else:
raise IndexError("Stack is empty")
def peek(self):
if not self.is_empty():
return self.stack[-1]
else:
raise IndexError("Stack is empty")
def is_empty(self):
return len(self.stack) == 0
def size(self):
return len(self.stack)
# 示例用法
stack = Stack()
stack.push(1)
stack.push(2)
stack.push(3)
print(stack.pop()) # 输出: 3
print(stack.peek()) # 输出: 2
print(stack.is_empty()) # 输出: False
print(stack.size()) # 输出: 2
```
这只是一个简单的示例,你可以根据实际需求进行修改和扩展。
阅读全文