How can I implement stacks in Python?

How can I implement stacks in Python?

You can implement a stack in Python using a list or by creating a custom class. Would you like to see both methods?

Yes, please show me how to do it with a list first.

Sure! You can use the append() method to push items onto the stack and pop() to remove them. Here's a simple example:

stack = []
stack.append(1)  # Push 1
stack.append(2)  # Push 2
print(stack)     # Output: [1, 2]
stack.pop()     # Pop the top item
print(stack)     # Output: [1]

Great! How would I implement it using a custom class?

You can create a Stack class like this:

class Stack:
    def __init__(self):
        self.items = []

    def push(self, item):
        self.items.append(item)

    def pop(self):
        return self.items.pop() if not self.is_empty() else None

    def is_empty(self):
        return len(self.items) == 0

stack = Stack()
stack.push(1)
stack.push(2)
print(stack.pop())  # Output: 2