add stack.py

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)}')