如何用Python将嵌套循环索引及结果按指定格式保存为CSV文件

原代码存在冗余的内层循环（for _ in range(2500)），这部分循环只是重复计算同一个res值，完全可以直接省略。我们的目标是遍历所有i,j,k,l,m,n的参数组合，计算对应的r1（即i+j+k+l+m+n），并将所有参数和结果导出为指定格式的CSV文件。

实现思路 #

1. 移除冗余循环：内层2500次循环重复计算相同的res，最终r1就是该res本身（所有重复值的最大值还是它自己），直接计算即可。
2. 简化嵌套循环：用itertools.product一次性生成所有参数的笛卡尔积，替代多层嵌套循环，代码更简洁高效。
3. 导出CSV：可选择Python内置的csv模块，或更便捷的pandas库完成数据导出。

代码实现 #

方法1：使用Python内置csv模块 #

import itertools
import csv
import numpy as np

# 定义各参数的取值范围
i_vals = range(100, 1500, 100)
j_vals = [round(x, 1) for x in np.arange(0, 5, 0.5)]  # 修正浮点数精度问题
k_vals = range(100, 1500, 100)
l_vals = [round(x, 1) for x in np.arange(0, 5, 0.5)]
m_vals = range(1000, 10000, 500)
n_vals = range(0, 20, 2)

# 存储所有参数与结果
data = []
# 生成所有参数组合并计算r1
for i, j, k, l, m, n in itertools.product(i_vals, j_vals, k_vals, l_vals, m_vals, n_vals):
    r1 = i + j + k + l + m + n
    data.append([i, j, k, l, m, n, r1])

# 写入CSV文件（制表符分隔，匹配示例格式）
with open('result.csv', 'w', newline='') as f:
    writer = csv.writer(f, delimiter='\t')
    writer.writerow(['i', 'j', 'k', 'l', 'm', 'n', 'r1'])  # 写入表头
    writer.writerows(data)

方法2：使用pandas（更简洁） #

import itertools
import pandas as pd
import numpy as np

# 定义各参数的取值范围
params = {
    'i': range(100, 1500, 100),
    'j': np.arange(0, 5, 0.5),
    'k': range(100, 1500, 100),
    'l': np.arange(0, 5, 0.5),
    'm': range(1000, 10000, 500),
    'n': range(0, 20, 2)
}

# 生成所有参数组合的DataFrame
df = pd.DataFrame(itertools.product(*params.values()), columns=params.keys())
# 计算r1列
df['r1'] = df.sum(axis=1)

# 导出为制表符分隔的CSV
df.to_csv('result.csv', sep='\t', index=False)

关键说明 #

• 浮点数精度：np.arange生成的浮点数可能存在微小误差，方法1中用round(x,1)修正，pandas会自动处理这类精度问题。
• 效率提升：原代码的冗余循环会导致计算量暴增，优化后直接生成参数组合，效率提升数个数量级。
• 格式调整：如果需要逗号分隔的标准CSV，只需将导出代码中的sep='\t'改为sep=','即可。

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