使用python的绘图库matplotlib绘制sigmoid激活函数,下面是具体的python代码(参考了简书中的添加xy坐标轴的代码)
import matplotlib.pyplot as plt
import numpy as np
import mpl_toolkits.axisartist as axisartist
def sigmoid(x):
return 1 / (1 + np.exp(-x))
def image():
# 使用x生成若干个的点集合
x = np.arange(-10, 11, 0.01)
y = sigmoid(x)
# 设置范围
plt.xlim(-10, 10)
plt.ylim(-1, 1)
# 下面是画出对应的xy坐标轴
# 创建画布
fig = plt.figure(figsize=(6, 6))
# 使用axisartist.Subplot方法创建一个绘图区对象ax
ax = axisartist.Subplot(fig, 111)
# 将绘图区对象添加到画布中
fig.add_axes(ax)
# 通过set_visible方法设置绘图区所有坐标轴隐藏
ax.axis[:].set_visible(False)
# ax.new_floating_axis代表添加新的坐标轴
ax.axis["x"] = ax.new_floating_axis(0, 0)
# 给x坐标轴加上箭头
ax.axis["x"].set_axisline_style("->", size=1.0)
# 添加y坐标轴,且加上箭头
ax.axis["y"] = ax.new_floating_axis(1, 0)
ax.axis["y"].set_axisline_style("-|>", size=1.0)
# 设置x、y轴上刻度显示方向
ax.axis["x"].set_axis_direction("top")
ax.axis["y"].set_axis_direction("right")
plt.plot(x, y, color="g", linewidth=3)
plt.show()
if __name__ == '__main__':
image()