data-structures/stack/stack.py

29 lines
549 B
Python
Raw Normal View History

2022-08-29 13:48:31 +00:00
# Create a stack
def create_stack():
stack = []
return stack
# Create an empty stack
def check_empty(stack):
return len(stack) == 0
# Add items to stack
def push(stack, item):
stack.append(item)
print(f'Pushed {item}')
# Remove an element from stack
def pop(stack):
if (check_empty(stack)):
return "stack empty"
return stack.pop()
# Driver code
stack = create_stack()
push(stack, str(1))
push(stack, str(2))
push(stack, str(3))
push(stack, str(4))
print(f'Popped: {pop(stack)}')
print(f'Stack: {str(stack)}')