Python中dict[str,list[MyClass]]字典添加元素后列表仅1个元素如何修复

问题根因 #

• 核心问题：判断None时使用了==而非Python规范的is，如果代码中存在自定义__eq__的逻辑，会导致None的判断结果不符合预期，这是代码始终无法进入if分支的直接原因。
• 潜在问题：items被定义为类属性，所有MyDict实例共享同一个字典，多实例场景下会出现数据串扰。
• 写法冗余：原代码每次通过update重新构造键值对、拼接生成新列表，不仅执行效率低，也增加了出错概率。

修复方案 #

最简写法（推荐） #

直接使用Python字典内置的setdefault方法，无需额外分支判断，一行即可实现需求：

class MyDict:
    # 改为实例属性，在初始化方法中单独初始化每个实例的字典
    def __init__(self):
        self.items: dict[str, list[MyClass]] = {}

    # other stuff omitted

    def addElement(self, p: str, c: MyClass):
        # 键不存在时自动设为空列表，返回对应列表后直接追加元素
        self.items.setdefault(p, []).append(c)

如果你的业务需求是将新元素追加到列表头部而非尾部，把append(c)替换为insert(0, c)即可。

保留原分支写法的修复版本 #

如果你需要保留原有打印日志的逻辑，可以按如下方式修改：

def addElement(self, p: str, c: MyClass):
    tmp: list[MyClass] = self.items.get(p)
    # 用is None判断空值，避免自定义相等性逻辑的干扰
    if tmp is None or len(tmp) == 0:
        print("key not exists, inserting")
        self.items[p] = [c]
    else:
        print("existing key, adding")
        # 直接操作原列表追加元素，无需重新构造列表和字典键值对
        tmp.append(c)

内容的提问来源于stack exchange，提问作者Finley Adams