如何在Julia中实现类似Python itertools.product的组合求和功能？

我有一个N-M维数组A，需要生成从每个子数组中各取一个元素的所有组合，并计算每个组合的和。示例：A=[[1,2],[3,4],[5,6]]，对应的组合为[1,3,5]、[1,4,5]、[1,4,6]、[2,3,5]、[2,3,6]、[2,4,5]、[2,4,6]，求和结果依次为9、10、11、10、11、11、12。

我已用Python的itertools.product实现该功能，代码如下：

import itertools

A = [[1,2], [3,4], [5,6]]
sums = ((vs, sum(vs)) for vs in itertools.product(*A))
for k, v in sums:
    print(v)
    print(k)

请问如何用Julia实现该功能？

解决方案 #

在Julia中，可以借助Iterators.product实现与Pythonitertools.product相同的笛卡尔积生成逻辑，以下是对应实现：

基础实现（与Python示例格式完全匹配）

using Iterators

A = [[1,2], [3,4], [5,6]]
# 生成所有子数组元素的笛卡尔积组合，用...展开数组参数
combinations = Iterators.product(A...)

for combo in combinations
    # 将元组转为数组，匹配Python示例中的列表格式
    combo_arr = collect(combo)
    sum_val = sum(combo_arr)
    println(sum_val)
    println(combo_arr)
end

简化版（直接操作元组）

如果不需要将组合转为数组，直接对元组求和即可，代码更简洁：

using Iterators

A = [[1,2], [3,4], [5,6]]
for combo in Iterators.product(A...)
    sum_val = sum(combo)
    println(sum_val)
    println(combo)
end

关键说明 #

• Iterators.product(A...)：和Python的itertools.product(*A)逻辑一致，生成所有子数组元素的笛卡尔积，每个组合以元组形式返回。
• collect(combo)：将元组转换为数组，与Python示例中的输出格式对齐。
• sum(combo)：Julia的sum函数支持直接对元组求和，无需额外转换。

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