如何基于可迭代对象元素向defaultdict(list)追加元素？Python代码调试

嘿，我来帮你排查这段代码的问题，然后实现你想要的按函数返回值分组的功能～

你的代码主要有两个核心问题：

• 全局字典的数据污染风险：你在函数外部定义了d = defaultdict(list)这个全局变量，要是多次调用group_by_value，之前的分组结果会一直留在这个字典里，导致后续结果完全错误。
• 迭代器遍历逻辑完全错误：
  • map(func, it)返回的是迭代器，这类对象只能被遍历一次，当你外层循环for i in result之后，这个迭代器就被耗尽了，后续再操作也拿不到数据。
  • 嵌套循环的逻辑完全偏离需求：外层遍历每个函数返回值，内层又遍历整个原可迭代对象，这会把所有元素都塞到每个键对应的列表里，根本不是按函数返回值分组的逻辑。

下面是修复后的完整代码，完全实现你想要的「键为函数返回值，列表为对应原元素」的结构：

from collections import defaultdict

def modn(n):
    return n % 3

def group_by_value(it, func):
    # 把字典创建移到函数内部，每次调用都生成新字典，避免数据残留
    result_dict = defaultdict(list)
    # 直接遍历原可迭代对象的每个元素
    for item in it:
        # 计算当前元素对应的分组键
        group_key = func(item)
        # 将元素添加到对应键的列表中
        result_dict[group_key].append(item)
    # 如果需要返回普通dict而非defaultdict，可以改成 return dict(result_dict)
    return result_dict

numbers = [1, 4, 5, 6, 8, 19, 34, 55]
print(group_by_value(numbers, modn))

执行后会输出符合预期的分组结果：

defaultdict(<class 'list'>, {1: [1, 4, 19, 34], 2: [5, 8, 55], 0: [6]})

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