numpy.savetxt保存itertools组合时报错：需1D/2D数组却得到0D数组

itertools.combinations() 函数返回的是惰性迭代器对象，不是结构化的序列容器。numpy 在接收该迭代器作为数组输入时，不会自动遍历迭代器读取内部的组合元组，而是会将整个迭代器对象识别为单个 0 维（标量）元素，生成的数组维度不符合 numpy.savetxt() 要求的 1D/2D 维度规则，因此抛出对应错误。

你可以通过以下代码验证该逻辑：

import itertools
import numpy as np

comb_iter = itertools.combinations(range(1,5), 3)
arr = np.array(comb_iter)
print(arr.ndim) # 输出为 0，符合报错提示的0D数组特征

将迭代器转换为列表后再传入numpy即可，转列表操作会完整加载所有组合元组，生成的嵌套序列可以被numpy自动解析为符合要求的二维数组：

# 简化场景示例
comb_list = list(itertools.combinations(range(1,5), 3))
# 实际业务场景替换为 range(1,91)、取5个参数即可
# comb_list = list(itertools.combinations(range(1,91), 5))
np.savetxt("file.csv", comb_list, delimiter=",", fmt="%d")

添加 fmt="%d" 配置可以保证输出的CSV内数字为整数格式，避免默认输出浮点数样式的内容。

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