如何展平SymPy生成的楼板几何场景组合嵌套输出？

针对你用SymPy的multiset_combinations和iproduct生成楼板几何场景时出现的嵌套结构（如([1000, 1500], 50)），可以通过以下两种方式得到扁平元组(1000, 1500, 50)：

一、事后转换现有嵌套结构 #

如果已经生成了嵌套的组合结果，直接通过解包语法批量转换即可：

示例代码 #

from sympy.utilities.iterables import multiset_combinations
from itertools import iproduct

# 定义参数范围（示例：长度单位毫米，200米=200000mm；厚度120cm=1200mm）
length_range = range(1000, 200001, 500)
thickness_range = range(50, 1201, 50)

# 生成原始嵌套结构
length_combs = multiset_combinations(length_range, 2)
nested_scenarios = iproduct(length_combs, thickness_range)

# 转换为扁平元组
flat_scenarios = [(*comb_list, thickness) for comb_list, thickness in nested_scenarios]

# 输出示例：(1000, 1500, 50)
print(flat_scenarios[0])

如果需要节省内存（场景数量大时），可以用生成器表达式替代列表推导式：

flat_scenarios = ((*comb_list, thickness) for comb_list, thickness in nested_scenarios)

二、从根源避免嵌套结构 #

跳过iproduct，直接用嵌套生成器/循环生成扁平元组，从源头避免嵌套：

示例代码 #

from sympy.utilities.iterables import multiset_combinations

# 同样定义参数范围
length_range = range(1000, 200001, 500)
thickness_range = range(50, 1201, 50)

# 直接生成扁平元组场景
flat_scenarios = (
    (*comb, thickness)
    for comb in multiset_combinations(length_range, 2)
    for thickness in thickness_range
)

# 输出示例：(1000, 1500, 50)
print(next(flat_scenarios))

这种写法逻辑更直观，无需额外转换步骤，同时保持了迭代器的惰性计算特性，适合处理大范围的场景遍历。

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