Python写入文件时如何将格式指定为可迭代对象适配多数组

我目前使用以下Python代码写入数据文件：

f = open(filename,'w')

for i in range(len(xarr)):
   f.write("%16.8e	%16.8e	%16.8e\n" %(xarr[i], y1[i], y2[i]))
f.close()

我希望将这段逻辑封装为函数，用于批量写入多个文件，但每次需要写入的数组数量（如y1、y2等）并不固定。请问如何将格式指定为可迭代对象，使其能根据数组数量动态调整？

解决方案 #

动态拼接格式字符串实现适配 #

核心思路是根据传入的数组数量自动生成对应格式占位符，结合zip批量提取对应位置的元素，代码实现如下：

def write_data(filename, xarr, *yarrs):
    # 动态生成格式字符串：x列占位符 + 对应数量的y列占位符 + 换行符
    fmt = "%16.8e" + "\t%16.8e" * len(yarrs) + "\n"
    # 使用with语句自动管理文件资源，无需手动close
    with open(filename, 'w') as f:
        # zip打包所有数组的对应元素，循环写入每一行
        for items in zip(xarr, *yarrs):
            f.write(fmt % items)

使用示例 #

# 测试数据
x = [1.0, 2.0, 3.0]
y1 = [4.0, 5.0, 6.0]
y2 = [7.0, 8.0, 9.0]
y3 = [10.0, 11.0, 12.0]

# 写入包含x+y1+y2的文件
write_data("two_y_data.txt", x, y1, y2)
# 写入包含x+y1+y2+y3的文件
write_data("three_y_data.txt", x, y1, y2, y3)

关键细节说明 #

• *yarrs用于接收任意数量的可变数组参数，完美适配不同数量的y数组需求
• 格式字符串通过"%16.8e" * len(yarrs)动态生成对应数量的y列占位符，和x列占位符拼接后得到完整格式
• zip(xarr, *yarrs)会将所有数组的第i个元素打包成元组，直接匹配格式字符串的参数要求
• with语句自动处理文件打开/关闭，避免手动操作可能导致的资源泄漏

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