【深度学习】论文画图

import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt

# 示例数据
data = {
    'L': ['L-4', 'L-4', 'L-4', 'L-8', 'L-8', 'L-8', 'L-16', 'L-16', 'L-16', 'L-24', 'L-24', 'L-24', 'L-32', 'L-32', 'L-32'],
    'Method': ['LSTM & LSTM', 'GPT & Table', 'LSTM & Table'] * 5,
    'Accuracy': [60, 50, 40, 70, 60, 50, 65, 55, 45, 68, 58, 48, 72, 62, 52]
}

# 创建 DataFrame
df = pd.DataFrame(data)

# 设置图形大小
fig, axs = plt.subplots(2, 3, figsize=(18, 12))  # 创建 2 行 3 列的子图

# 自定义颜色
palette = {
    'LSTM & LSTM': 'orange',
    'GPT & Table': 'yellow',
    'LSTM & Table': 'green'
}

# 绘制每个子图
for i, ax in enumerate(axs.flat):
    sns.barplot(data=df, x='L', y='Accuracy', hue='Method', palette=palette, width=0.3, ax=ax)
    ax.set_ylabel('Accuracy (%)')
    ax.set_ylim(0, 100)
    ax.set_title(f'Subplot {i + 1}', fontsize=14, loc='center', pad=20)  # 每个子图的标题
    ax.legend(title='Methods', loc='upper right')  # 图例位置
    ax.spines['right'].set_visible(False)  # 去掉右侧坐标轴
    ax.spines['top'].set_visible(False)  # 去掉上方坐标轴
    ax.xaxis.set_ticks_position('none')  # 去掉 x 轴的刻度
    ax.yaxis.set_ticks_position('none')  # 去掉 y 轴的刻度

# 调整布局
plt.tight_layout()
plt.show()  # 显示所有子图