将多个子图的legend放在一起

matlab的旋律

在评价模型当中，经常需要到多模型多指标同时显示，如何实现多个子图的legend放在一起呢?这是可以使用画布figure的suptitle，具体实现代码如下：

1. import numpy as np
   
2. import pickle
   
3. import matplotlib.pyplot as plt# plt 用于显示图片
   
4. 
   
5. 
   
6. file = ['scnn', 'scg', 'scga']
   
7. color = ['-r', '-g', '-b']
   
8. 
   
9. # plt.style.use('dark_background')
   
10. fig = plt.figure()
    
11. ax1 = fig.add_subplot(121)
    
12. ax2 = fig.add_subplot(122)
    
13. fig.suptitle('')
    
14. 
    
15. for k in range(len(file)):
    
16.     with open(file[k] + '.txt', 'rb') as file_pi:
    
17.            ggn_history = pickle.load(file_pi)
    
18.     y_loss = ggn_history['loss']
    
19.     y_err = ggn_history['accuracy']
    
20.     ax1.plot(y_err[0:20], color[k])
    
21.     ax2.plot(y_loss[0:20], color[k])
    
22. 
    
23. 
    
24. ax1.set_title('accuracy')
    
25. ax2.set_title('loss')
    
26. ax1.set_xlabel('epochs')
    
27. ax2.set_xlabel('epochs')
    
28. fig.legend(('scn', 'scg', 'scga'), ncol=3, loc='upper center')
    
29. plt.savefig('loss_acc.png')
    
30. plt.show()
    

复制代码