近日做作業需要畫PR圖,手繪數據量略大,故使用plt進行繪圖
參考該博主代碼:https://blog.csdn.net/weixin_38950569/article/details/97817146
在此基礎上增加了自行設置刻度功能(由於數據量的問題,不自己設置刻度,畫出來的圖看上去有點奇怪)
import matplotlib.pyplot as plt precision = [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 15/16, 16/17, 16/18, 17/19, 18/20, 18/21, 19/22, 19/23, 20/24] recall = [1/80, 2/80, 3/80, 4/80, 5/80, 6/80, 7/80, 8/80, 9/80, 10/80, 11/80, 12/80, 13/80, 14/80, 15/80, 15/80, 16/80, 16/80, 16/80, 17/80, 17/80, 18/80, 18/80, 19/80] plt.plot(recall, precision, linewidth=3, color='b') # 將列表傳遞給plot,並設置線寬,設置顏色,默認為藍色 plt.title("P-R", fontsize=24, color='r') # 設置標題,並給定字號,設置顏色 plt.xlabel("Recall", fontsize=14, color='g') # 設置軸標題,並給定字號,設置顏色 plt.ylabel("Precision", fontsize=14, color='g') a = 10 plt.xticks(recall) plt.yticks([1/a, 2/a, 3/a, 4/a, 5/a, 6/a, 7/a, 8/a, 9/a, 10/a]) # 設置x和y軸的刻度 plt.tick_params(axis='both', labelsize=14) # 設置刻度標記的大小 plt.show() # 顯示