Python绘制sigmoid激活函数

使用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()