畫(huà)圖
pyplot.plot() 的官方文檔解釋?zhuān)?a target="_blank">鏈接
1.1 畫(huà)二維平面圖
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
x = np.arange(0, 10, 3)
y = np.arange(0, 10, 3)
print(x)
plt.plot(x, y, '*g--')
plt.show()
以上代碼是畫(huà) 二維平面圖宇驾，python plt.plot(x, y, '*g--') 這里的x 表示橫坐標(biāo)刨裆， y 表示縱坐標(biāo)艾杏， 字符串 python '*g--' 其中 * 表示每個(gè)點(diǎn)用 五角星表示昔穴， g 表示線段的顏色為綠色， -- 表示線段的類(lèi)型则北，如果需要更換 點(diǎn)的形狀邢隧、 顏色、線段可以參考官方文檔隶糕。
1.2 同一個(gè)坐標(biāo)系里畫(huà)多條線段
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
x = np.arange(0, 10, 3)
y1 = x
y2 = x**2
plt.plot(x, y1, '*g--', y2, '^b-')
plt.legend(['y=x', '$y=x^2$'], loc='upper right') # 顯示每條線段的解釋?zhuān)?$$ 里是 LaTeX語(yǔ)句
plt.show()
注意：默認(rèn)如果使用漢字瓷产，顯示是不正常的，這里重新設(shè)置顯示字體即可枚驻，代碼如下：
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
x = np.arange(0, 10, 3)
y1 = x
y2 = x**2
# 正常顯示中文
from pylab import mpl
mpl.rcParams['font.sans-serif'] = ['SimHei']
plt.xlabel('自變量') # 若是使用 漢字濒旦，則顯示出錯(cuò)
plt.ylabel('因變量')
plt.plot(x, y1, '*g--', y2, '^b-')
plt.legend(['y=x', '$y=x^2$'], loc='upper right') # 顯示每條線段的解釋?zhuān)?$$ 里是 LaTeX語(yǔ)句
plt.show()
1.3 封裝畫(huà)圖的函數(shù)
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
def show_error(x_name:list, precision:list, recall:list, f1:list, AUC:list, path):
plt.plot(x_name, precision, 'or-', recall, '^g-', f1, '*b-', AUC, '.y-.')
plt.legend(['precision', 'recall', 'f1', 'AUC'], loc='upper right')
plt.savefig(path, dpi=300)
plt.show()
pass
x_name = ['model1', 'model2', 'model3', 'model4']
y1 = [0.5, 0.6, 0.8, 0.9]
y2 = [0.6, 0.6, 0.85, 0.92]
y3 = [0.7, 0.66, 0.82, 0.91]
y4 = [0.8, 0.7, 0.82, 0.84, 0.93]
show_error(x_name,y1 ,y2 ,y3 , y4, r'test.png')
這里每一條線段代表一個(gè)評(píng)價(jià)指標(biāo), 橫坐標(biāo)代表不同的模型，縱坐標(biāo)代表指標(biāo)的大小再登，每一條折線尔邓，代表該不同模型的相同評(píng)價(jià)指標(biāo)的值。通過(guò)折線圖可以直觀的觀察出模型的好壞锉矢。
