### Python 类中 `__getitem__` 方法的用法 在 Python 中，特殊方法（也称为魔术方法）允许我们自定义类的行为以适应特定需求。其中，`__getitem__` 是一种用于支持通过键或索引来访问对象内部数据的方法。具体来说，当尝试使用方括号语法（如 `obj[key]`）访问某个对象的内容时，会自动调用此方法[^3]。 #### 1. **基本功能** 如果在一个类中实现了 `__getitem__` 方法，则该类的对象就可以像字典或者列表一样按索引/键取值。这使得我们可以构建更加灵活的数据容器类型[^3]。 ```python class MyDataContainer: def __init__(self, data): self.data = data def __getitem__(self, key): return self.data[key] container = MyDataContainer([10, 20, 30]) print(container[1]) # 输出: 20 ``` #### 2. **处理切片操作** 当传递给 `__getitem__` 的不是一个单一整数而是切片对象 (slice object) 时，我们需要特别注意如何解析这种输入。下面的例子展示了如何让我们的类能够响应切片请求[^3]。 ```python class SliceableList: def __init__(self, items): self.items = items def __getitem__(self, index): if isinstance(index, slice): start, stop, step = index.indices(len(self.items)) return [self.items[i] for i in range(start, stop, step)] elif isinstance(index, int): return self.items[index] else: raise TypeError("Invalid argument type.") sl = SliceableList(['a', 'b', 'c', 'd']) print(sl[:2]) # 输出: ['a', 'b'] print(sl[::-1]) # 输出: ['d', 'c', 'b', 'a'] ``` #### 3. **模拟映射型数据结构** 我们还可以利用 `__getitem__` 来模仿字典的行为，从而创建自定义的映射型数据结构[^3]。 ```python class CustomDict: def __init__(self, dictionary=None): if dictionary is None: self.dict = {} else: self.dict = dict(dictionary) def __getitem__(self, key): try: return self.dict[key] except KeyError: return f"No such key '{key}' exists." cd = CustomDict({'name': 'Alice', 'age': 25}) print(cd['name']) # 输出: Alice print(cd['height']) # 输出: No such key 'height' exists. ``` #### 4. **错误处理与边界条件** 在实际开发过程中，合理地捕获并反馈潜在的异常情况是非常重要的。比如，在上面例子的基础上进一步完善错误提示信息可以帮助使用者更快定位问题所在[^3]。 --- ### 示例代码展示 这里给出一个完整的案例，演示了如何结合多种特性设计一个具有高级特性的序列化容器： ```python class AdvancedSequence: def __init__(self, elements=None): if elements is None: self.elements = [] else: self.elements = list(elements) def append(self, element): """添加新元素""" self.elements.append(element) def extend(self, iterable): """扩展多个元素""" self.elements.extend(iterable) def __len__(self): return len(self.elements) def __getitem__(self, idx): """ 支持单个索引获取、负数索引以及切片操作。 若超出范围则抛出 IndexError 异常。 """ if isinstance(idx, slice): return self.__class__(self.elements[idx]) else: return self.elements[idx] # 测试部分 seq = AdvancedSequence(range(1, 6)) # 初始化为 [1, 2, 3, 4, 5] print(seq[-1]) # 获取最后一个元素 -> 5 sub_seq = seq[::2] # 切片得到偶数位置上的子集 for num in sub_seq: print(num) # 应打印 1, 3, 5 各占一行 ``` 在这个示例里，不仅实现了基础的功能还加入了额外的支持，例如链式调用的可能性(`AdvancedSequence`)和更复杂的迭代器协议遵循等。 --- ### 总结 - `__getitem__` 能够让我们轻松实现类似数组或哈希表那样的存取语义。 - 对于复杂查询模式下的适配工作，需考虑更多细节诸如切片表达式的兼容性和健壮性保障等问题。 - 结合其他魔法函数一起运用可极大增强程序表现力的同时保持简洁优雅的设计风格。 ---