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__ - 知乎

你可能感兴趣的:(Python,python,开发语言,__getitem__)