add stack.py

This commit is contained in:
Aditya 2022-08-29 19:18:31 +05:30
commit c050979213

28
stack/stack.py Normal file
View file

@ -0,0 +1,28 @@
# 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)}')