如何用Julia的Iterators.product生成带组合函数值的DataFrame？

问题分析 #

你当前的代码中：

• firstdataframe 正确生成了笛卡尔积的组合结构，但缺少计算列；
• seconddataframe 出错是因为 Iterators.product 返回的是多维迭代器，直接遍历得到的是多维数组结构，转DataFrame后会把数组作为单元格值，而非展开成行。

优雅实现方案 #

方案1：先生成组合DataFrame，再追加计算列 #

这是最直观的方式，先得到正确的组合结构，再通过行运算添加计算列：

using DataFrames

numberset = [3, 7]
k = 2

# 生成笛卡尔积组合的DataFrame
df = DataFrame(Iterators.product(Iterators.repeated(numberset, k)...), :auto)
# 可选：将默认列名改为n1、n2...
rename!(df, ["n$i" for i in 1:k])
# 添加求和列（可替换为任意自定义函数）
df.sum = sum.(eachrow(df))

运行后得到预期结果：

4×3 DataFrame
 Row │ n1     n2     sum
     │ Int64  Int64  Int64
─────┼──────────────────────
   1 │     3      3      6
   2 │     7      3     10
   3 │     3      7     10
   4 │     7      7     14

方案2：一次性构造含计算列的DataFrame #

先将多维笛卡尔积扁平化，再生成包含组合与计算值的行数据：

using DataFrames

numberset = [3, 7]
k = 2

# 将多维笛卡尔积转为一维元组集合
flat_combinations = vec(collect(Iterators.product(Iterators.repeated(numberset, k)...)))
# 生成每行数据：组合元组 + 计算值
rows = [tuple(c..., sum(c)) for c in flat_combinations]
# 构造DataFrame并指定列名
df = DataFrame(rows, ["n$i" for i in 1:k] ∪ ["sum"])

额外说明 #

• 如果你需要的是无顺序的组合（如(3,7)与(7,3)视为同一组），而非笛卡尔积，可使用Combinatorics.jl包的combinations函数替代Iterators.product；
• 自定义计算函数可替换sum为任意逻辑，比如prod（乘积）、自定义匿名函数等。

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