如何仅为单个绘图显示图例？多数据集同图绘制技术咨询

我来帮你搞定这个在同一张图里绘制两个数据集的需求～先看看你现有代码里的小问题，再给你两种实用的实现方案：

先修正代码里的小问题 #

你的代码里有个Python大小写敏感的坑：循环里写的for fname in Filelist:应该改成for fname in filelist，不然会报变量未定义的错误。另外，读取15个文件时需要把数据存储起来，才能方便后续绘图操作。

完整实现方案 #

这里提供两种常见的绘图思路，你可以根据自己的需求选择：

方式一：逐个绘制15个文件的数据，叠加单个文件的基准曲线 #

这种方式适合观察15个文件数据的离散分布，和单个文件做直观对比：

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
%matplotlib notebook

# 生成15个目标文件的路径列表
filelist = []
for i in range(1, 16):
    filelist.append("/Users/Hrihaan/Desktop/Code/A_B_%s.txt" % i)

# 读取单个基准文件的数据
data_single = pd.read_table('/Users/Hrihaan/Desktop/Code/A_B_0.txt', dtype=float, header=None, sep='\s+').values

# 创建绘图窗口
plt.figure(figsize=(10,6))

# 循环读取并绘制15个文件的数据（设置低透明度避免重叠太乱）
for fname in filelist:
    data = pd.read_table(fname, dtype=float, header=None, sep='\s+').values
    # 假设数据是两列：第一列x轴，第二列y轴，可根据实际情况调整索引
    plt.plot(data[:,0], data[:,1], alpha=0.3, label=f'File {fname.split("_")[-1].split(".")[0]}')

# 绘制单个基准文件的数据，用醒目的样式突出
plt.plot(data_single[:,0], data_single[:,1], color='#ff3333', linewidth=2, label='Benchmark (A_B_0.txt)')

# 添加图表必要元素
plt.xlabel('X Axis Label')
plt.ylabel('Y Axis Label')
plt.title('Benchmark vs 15 Experimental Datasets')
plt.legend(bbox_to_anchor=(1.05, 1), loc='upper left')  # 把图例放在图外避免遮挡
plt.tight_layout()
plt.show()

方式二：绘制15个文件的统计特征（均值±标准差），对比单个文件 #

如果15个文件是重复实验的数据，更适合展示统计规律：

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
%matplotlib notebook

# 生成15个目标文件的路径列表
filelist = []
for i in range(1, 16):
    filelist.append("/Users/Hrihaan/Desktop/Code/A_B_%s.txt" % i)

# 读取单个基准文件的数据
data_single = pd.read_table('/Users/Hrihaan/Desktop/Code/A_B_0.txt', dtype=float, header=None, sep='\s+').values

# 存储15个文件的y轴数据
all_y_data = []
for fname in filelist:
    data = pd.read_table(fname, dtype=float, header=None, sep='\s+').values
    all_y_data.append(data[:,1])  # 假设y值在第二列，可按需调整

# 转换为numpy数组，计算均值和标准差
all_y_np = np.array(all_y_data)
mean_y = np.mean(all_y_np, axis=0)
std_y = np.std(all_y_np, axis=0)
x_axis = data[:,0]  # 假设所有文件的x轴一致，取任意一个文件的x轴即可

# 创建绘图窗口
plt.figure(figsize=(10,6))

# 绘制15个文件的均值曲线，并用标准差填充误差范围
plt.plot(x_axis, mean_y, color='#3366ff', linewidth=2, label='15 Files Mean')
plt.fill_between(x_axis, mean_y - std_y, mean_y + std_y, color='#3366ff', alpha=0.2)

# 绘制单个基准文件的数据
plt.plot(data_single[:,0], data_single[:,1], color='#ff3333', linewidth=2, linestyle='--', label='Benchmark (A_B_0.txt)')

# 添加图表必要元素
plt.xlabel('X Axis Label')
plt.ylabel('Y Axis Label')
plt.title('Benchmark vs 15 Files (Mean ± Standard Deviation)')
plt.legend()
plt.tight_layout()
plt.show()

小提示 #

• 如果你的数据列数和假设的不一样（比如只有一列数据，或者x轴在其他列），只需要调整代码里的data[:,0]、data[:,1]索引即可。
• %matplotlib notebook是交互式绘图模式，如果需要静态图可以换成%matplotlib inline。

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