这段代码能够解决plt.legend里数学公式输出和设置字体问题,简单说就是
font={
'family':'Times New Roman',
'weight':'light'
}
plt.legend(['relu:$\mathdefault{f(x) \equal max(0,x)}$','sigmoid:$\mathdefault{f(x)=\\frac{1}{1+e^{-x}}}$'],loc='upper left',prop=font)
核心部分是设置公式字体$\mathdefault{}$
和打分号要写成$\\frac{}{}$
不然\f会被识别成转义符号
这个应用性比较强,拿来跑运行即可,如有其它需求稍加修改就好,就不多解释了,有不懂的可以评论区或私聊问一下,上网搜也搜得到。
import matplotlib.pyplot as plt
from numpy import *
def relu(x):
y = x.copy()
y[y<0] = 0
return y
def sigmoid(x):
return 1.0 / (1.0 + exp(-x))
if __name__=='__main__':
x=np.arange(-2,2,0.01)
fig = plt.figure()
#移动y轴值中间,擦除顶端和右端边界线
ax = fig.add_subplot()
ax.spines['top'].set_color('none')
ax.spines['right'].set_color('none')
ax.spines['left'].set_position(('data', 0))
ax.spines['bottom'].set_position(('data', 0))
y_relu=relu(x)
y_sigmoid=sigmoid(x)
plt.tick_params(labelsize=14) # 刻度字体大小14
plt.plot(x,y_relu,'yellow')
plt.plot(x,y_sigmoid,'green')
font={
'family':'Times New Roman',
'weight':'light'
}
plt.legend(['relu:$\mathdefault{f(x) \equal max(0,x)}$','sigmoid:$\mathdefault{f(x)=\\frac{1}{1+e^{-x}}}$']
,loc='upper left',prop=font)
plt.tight_layout()#自动调整子图参数,使之填充整个图像区域
plt.show()