commit c050979213e4e0846bc599d067134f3813ddd9ed Author: Aditya Date: Mon Aug 29 19:18:31 2022 +0530 add stack.py diff --git a/stack/stack.py b/stack/stack.py new file mode 100644 index 0000000..e1295d3 --- /dev/null +++ b/stack/stack.py @@ -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)}')