在python中重载[]运算符？

class MyClass:
    def __init__(self, data):
        self.data = data

    def __getitem__(self, index):
        return self.data[index]

my_obj = MyClass([1, 2, 3, 4, 5])
print(my_obj[2])  # 输出：3

class MyClass:
    def __init__(self, data):
        self.data = data

    def __getitem__(self, index):
        if isinstance(index, slice):
            return self.data[index.start:index.stop:index.step]
        else:
            return self.data[index]

my_obj = MyClass([1, 2, 3, 4, 5])
print(my_obj[1:4])  # 输出：[2, 3, 4]