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 <module>
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__ – 知乎
来源:思考实践