Python中的__getitem__方法|极客笔记

Python中的`getitem`方法

Python中的<code>__getitem__</code>方法” title=”Python中的<code>__getitem__</code>方法” /></p>
<p>在Python中，<code>__getitem__</code>方法是一个特殊方法，用于访问对象的元素。通过重写<code>__getitem__</code>方法，我们可以自定义类的行为，使其支持像列表、字典等内置容器一样使用索引或键来访问元素。</p>
<h2 id= 实现__getitem__方法

要实现__getitem__方法，我们需要在类中定义该方法，并且指定希望通过索引或键访问的元素。接下来让我们通过几个示例来演示如何使用__getitem__方法。

示例一：自定义List类

class MyList:
    def __init__(self, *args):
        self.data = list(args)

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

    def __len__(self):
        return len(self.data)

my_list = MyList(1, 2, 3, 4, 5)
print(my_list[2])

运行结果：

在上面的示例中，我们自定义了一个List类MyList，并在其__getitem__方法中实现了通过索引访问元素的功能。当我们创建一个MyList对象并访问索引为2的元素时，会返回数字3，即第三个元素。

示例二：自定义Dict类

class MyDict:
    def __init__(self, *args, **kwargs):
        self.data = dict(*args, **kwargs)

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

    def __len__(self):
        return len(self.data)

my_dict = MyDict({'a': 1, 'b': 2, 'c': 3})
print(my_dict['b'])

运行结果：

在上面的示例中，我们自定义了一个Dict类MyDict，并在其__getitem__方法中实现了通过键访问元素的功能。当我们创建一个MyDict对象并访问键为’b’的元素时，会返回数字2。