Python中如何对itertools.combinations的组合结果执行自定义函数

嘿，我来帮你理清楚这个问题～你说打印时只返回对象内存地址，这大概率是因为你的迭代逻辑返回的是生成器/迭代器对象（比如Python里itertools.combinations就是这类），它不会一次性把所有组合都加载到内存里，而是按需生成，所以直接打印才会显示内存地址。

其实完全不用把所有组合都存下来，直接遍历这个迭代器，就能逐个拿到单个组合，然后传入你的自定义函数执行就行。给你举个具体的例子：

首先假设你的自定义函数是这样的（你可以根据自己的需求修改）：

def my_custom_function(combination):
    # 这里写你的业务逻辑，比如计算组合的总和，或者做其他处理
    print(f"正在处理组合：{combination}")
    # 举个例子，返回组合元素的乘积
    product = 1
    for num in combination:
        product *= num
    return product

然后你的组合生成逻辑（比如用itertools生成1到11的整数组合，这里的k你可以根据自己那39种组合的实际长度调整）：

import itertools

# 生成1到11的整数组合，这里假设是选k个元素的组合，你替换成对应k值即可
combo_iterator = itertools.combinations(range(1, 12), k=3)  # 举个例子，实际k要匹配你的39种组合

# 遍历迭代器，逐个处理每个组合
for single_combo in combo_iterator:
    # 把单个组合传入自定义函数
    func_result = my_custom_function(single_combo)
    # 如果需要，还可以对函数返回的结果做后续操作
    # print(f"函数返回结果：{func_result}")

几个关键细节： #

• 迭代器的核心就是按需生成单个元素，所以for循环里的single_combo每次就是一个完整的组合（比如(2,5,7)这样的元组，要是你需要列表格式，转成list(single_combo)就行）。
• 别把迭代器转成列表（比如list(combo_iterator)），这样会一次性把所有组合加载到内存，完全没必要，遍历迭代器反而更高效，尤其是组合数量多的时候。
• 如果只是想测试单个组合（比如先拿第一个试试），可以用next()函数直接从迭代器里取：

  first_combo = next(combo_iterator)
  my_custom_function(first_combo)
  

  要是想拿下一个，再调用一次next()就行，不过要注意迭代器是单向的，取过的元素就没法再取啦。

如果你的迭代逻辑是自己写的生成器函数（用了yield关键字），处理方式也是一模一样的，直接遍历或者用next()取单个元素传入函数就好。

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