python 类中的 __getitem__方法

经常会遇到python类中遇到一些特殊的方法，最常见的就是__ini__方法，想必大家都遇到过，今天来谈谈另外一种非常好用的类方法--__getitem__

来点干货， __getitem__的作用是什么呢？说白了就是类中一个特殊方法，类对象可以像字典对象那样根据key取值（dict['key']），如类对象Object['key']，系统会自动调用__getitem__方法(所以你在看pytorch里面Dataloader的时候没有发现主函数直接调用__getitem__，原来是系统自动调用)，然后返回该方法定义return值，说到这里，有一点需要强调哦， 字典取值时key不存在，会抛出异常，而类对象的key是否存在，都会调用__getitem__方法并返回其规定的值，下面进入实例。

代码

test = {
    'key1': 'value1',
    'key2': 'value2',
    'key3': 'value3',
    'key4': 'value4'
}
 
class Element:
    """获取元素"""
    def __init__(self):
        self.data = test
 
    def __getitem__(self, item):
        """获取属性"""
        data = self.data.get(item, '')
        if data:
            return data
        raise Exception('关键字不存在')
 
 
if __name__ == '__main__':
    print(Element()['key1'])
    
    print(Element()['key2'])
    print(Element()['key3'])
    print(Element()['key4'])
    print(Element()['key0'])

输出

(base) fight@fight-ZHENGJIUZHE-REN9000K-34IMZ:~/Desktop/Autoformer/Autoformer-main$ python details.py
value1
value2
value3
value4
Traceback (most recent call last):
  File "/home/fight/Desktop/Autoformer/Autoformer-main/details.py", line 27, in
    print(Element()['key0'])
  File "/home/fight/Desktop/Autoformer/Autoformer-main/details.py", line 18, in __getitem__
    raise Exception('关键字不存在')
Exception: 关键字不存在

参考资料

python 类中的 __getitem__方法_格度的博客-CSDN博客___getitem__ python

让你秒懂Python 类特殊方法__getitem__ - 知乎