__setitem__

setitem 是 Python 中的一个特殊方法，用于定义对象的索引赋值行为。它在你使用索引语法（例如 obj[key] = value）设置对象的元素时被自动调用。通过重载 setitem 方法，你可以自定义对象在处理索引赋值时的行为。

语法

class MyClass:
    def __setitem__(self, key, value):
        # 自定义的索引赋值逻辑
        pass

参数

self: 对象实例本身。
key: 要设置的索引键。
value: 要设置的值。

示例代码

1. 基本示例

下面是一个简单的示例，展示如何重载 setitem 方法：

class MyList:
    def __init__(self):
        self.data = []

    def __setitem__(self, index, value):
        print(f"Setting index {index} to {value}")
        self.data.insert(index, value)

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

my_list = MyList()
my_list[0] = 10  # 输出: Setting index 0 to 10
print(my_list[0])  # 输出: 10

在这个示例中，每次使用索引赋值时都会打印一条消息，并将值插入到 data 列表中。

2. 实现自定义字典

你可以使用 setitem 方法来实现一个自定义字典：

class MyDict:
    def __init__(self):
        self.data = {}

    def __setitem__(self, key, value):
        print(f"Setting key {key} to {value}")
        self.data[key] = value

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

    def __delitem__(self, key):
        print(f"Deleting key {key}")
        del self.data[key]

my_dict = MyDict()
my_dict['a'] = 1  # 输出: Setting key a to 1
print(my_dict['a'])  # 输出: 1
del my_dict['a']  # 输出: Deleting key a

3. 添加验证逻辑

你可以使用 setitem 方法来添加索引赋值的验证逻辑：

class PositiveList:
    def __init__(self):
        self.data = []

    def __setitem__(self, index, value):
        if value < 0:
            raise ValueError("Value cannot be negative")
        self.data.insert(index, value)

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

positive_list = PositiveList()
positive_list[0] = 10  # 正常设置
print(positive_list[0])  # 输出: 10

try:
    positive_list[1] = -5  # 将引发 ValueError
except ValueError as e:
    print(e)  # 输出: Value cannot be negative

注意事项

与 getitem 和 delitem 配合使用：
通常，setitem 会与 getitem 和 delitem 一起使用，以实现完整的索引操作。
性能影响：
重载 setitem 可能会影响性能，特别是在频繁进行索引赋值操作的情况下。因此，应谨慎使用。
避免无限递归：
在 setitem 方法内部设置元素时，应该直接操作底层数据结构，而不是使用索引赋值语法，以避免无限递归调用。

总结

setitem 是一个强大的工具，允许你自定义对象在处理索引赋值时的行为。通过重载 setitem，你可以添加验证逻辑、日志记录或其他自定义行为。然而，使用 setitem 时需要注意避免无限递归调用，并考虑到可能的性能影响。合理使用 setitem 可以使你的代码更加灵活和强大。