def getitem(self, idx)

最新推荐文章于 2024-09-06 00:09:57 发布

Mr_Happy_Li

最新推荐文章于 2024-09-06 00:09:57 发布

阅读量183

点赞数 4

文章标签： python 特殊方法

本文链接：https://blog.csdn.net/Mr_Happy_Li/article/details/141674223

版权

def getitem(self, idx):

这个方法用于定义对象的索引操作，即通过索引值获取对象中的元素。当你使用索引操作符[]来获取对象的元素时，实际上是调用了对象的__getitem__方法，并传入相应的索引值。

这个方法通常用于自定义数据集类中，用于按索引获取数据集中的样本。

调用方式：obj[idx]，其中obj是类的对象，idx是索引值。
代码：

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

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

# 创建自定义数据集对象
my_data = [1, 2, 3, 4, 5]
dataset = MyDataset(my_data)

# 通过索引获取数据集中的元素
element = dataset[2]
print("Element:", element)