Julia 中的 Itertools

Question

我有数组N-M维数组-A。我需要从每个字符串生成一个元素的组合并确定它们的总和。

例如，A=[[1,2],[3,4],[5,6]]。 k=[1,3,5],[1,4,5],[1,4,6],[2,3,5],[2,3,6],[2,4,5], [2,4,6].

因此 k 对于每个组合都有三个索引。总和=9,10,11,10,11,11,12。 使用Python，我可以用

做到这一点

sums = ((vs,sum(vs)) for vs in itertools.product(*A))
for k,v in sums:
    print v
    print k

如何使用 Julia 做到这一点？

Answer 1

julia> A=[[1,2],[3,4],[5,6]]; vec([(sum(v), v) for v in Iterators.product(A...)])
8-element Vector{Tuple{Int64, Tuple{Int64, Int64, Int64}}}:
 (9, (1, 3, 5))
 (10, (2, 3, 5))
 (10, (1, 4, 5))
 (11, (2, 4, 5))
 (10, (1, 3, 6))
 (11, (2, 3, 6))
 (11, (1, 4, 6))
 (12, (2, 4, 6))

看起来很接近。