如何将多曲线的x与y数据导出为txt文件的相邻列格式？

问题描述 #

现有绘制三条线性曲线的Python代码，需要将x轴数据与三条曲线的y轴数据导出到file.txt文件中，要求数据以相邻列的形式呈现，格式如下：

x1    y1    y2    y3

0     0     1     2 

1     2     3     4

2     4     5     6

解决方案 #

可以先统一收集所有数据，再按格式写入文件，修改后的完整代码如下：

import numpy as np
import matplotlib.pyplot as plt

# 定义参数与基础数据
z_points = np.array([0, 1, 2])
x_points = np.array([0, 1, 2])

# 批量计算所有z对应的y值
y_all = []
for z in z_points:
    y_points = 2 * x_points + z  # 利用numpy向量运算，无需逐个x循环
    y_all.append(y_points)

# 转置y数据，使每行对应同一个x的所有y值
y_all = np.array(y_all).T
# 拼接x列与所有y列，形成完整数据集
data = np.hstack([x_points.reshape(-1, 1), y_all])

# 写入文件
with open('file.txt', 'w') as f:
    # 写入表头
    f.write('x1    y1    y2    y3\n\n')
    # 逐行写入数据，用格式化字符串保证列对齐
    for row in data:
        f.write(f"{row[0]:<5}{row[1]:<5}{row[2]:<5}{row[3]:<5}\n")

# 保留原绘图逻辑
for z in z_points:
    y_points = 2 * x_points + z
    plt.plot(x_points, y_points, 'bo-', markersize=5)

plt.xlabel('x')
plt.ylabel('f(x)')
plt.grid(True, which='major', linestyle='-', linewidth='1', color='black')
plt.minorticks_on()
plt.grid(True, which='minor', linestyle=':', linewidth='1', color='gray')
plt.xlim([0, 5])
plt.ylim([0, 10])
plt.legend(['z[0]','z[1]','z[2]'])
plt.show()

关键步骤说明 #

• 数据收集：利用numpy的向量运算批量计算y值，替代原代码中嵌套循环，提升效率
• 数据整理：转置y数据矩阵，确保每行对应同一个x的三条曲线y值；将x列与y列拼接成符合要求的结构
• 文件写入：使用with open上下文管理器确保文件正确关闭；通过格式化字符串{:<5}实现左对齐的列格式，保证数据排版整齐

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