python matplotlib绘图——bar，两个柱子并列

Posted 2021-07-22 小乖乖的臭坏坏

import pandas as pd

read_csv = pd.read_csv('data.csv')

data_origin = read_csv['origin']
data_origin = data_origin.values
data_improve = read_csv['improve']
data_improve = data_improve.values
print(data_origin)

import matplotlib.pyplot as plt
import numpy as np
labels = ['Group 1', 'Group 2', 'Group 3', 'Group 4', 'Group 5']
x = np.arange(len(labels))
width = 0.35

fig, ax = plt.subplots()
rects1 = ax.bar(x-width/2, data_origin, width, label='Origin attention')
rects2 = ax.bar(x+width/2, data_improve, width, label='Improved attention')

ax.set_ylabel('MSE Loss')
ax.set_xticks(x)
ax.set_xticklabels(labels)
ax.legend()

fig.tight_layout()

plt.savefig('figure15.png')
plt.savefig('figure15.svg')
plt.close()

运行结果：