Matplotlib画图自留！！

心想市成

已于 2023-03-18 12:11:45 修改

阅读量84

点赞数 1

文章标签： matplotlib Powered by 金山文档

于 2023-03-18 12:00:05 首次发布

本文链接：https://blog.csdn.net/lwk431085/article/details/129634824

版权

本文展示了如何使用Python的numpy和matplotlib库来绘制sigmoid和swish函数图形，特别地，调整了线条颜色、类型和宽度，并将图像保存为PNG文件。代码中定义了中央轴的样式，使得图表更加简洁明了。

摘要由CSDN通过智能技术生成

另外附Swich绘图并输出保存为图片

#颜色设置：color='g'
#线的类型：linestyle='-'
#线的粗细:linewidth=5

import numpy as np
import matplotlib.pyplot as plt


def get_central_ax():
    ax = plt.gca()  # get current axis 获得坐标轴对象

    ax.spines['right'].set_color('none')
    ax.spines['top'].set_color('none')

    ax.xaxis.set_ticks_position('bottom')
    ax.yaxis.set_ticks_position('left')
    ax.spines['bottom'].set_position(('data', 0))
    ax.spines['left'].set_position(('data', 0))

    return ax
def sigmoid(x):
    return 1 / (1 + np.exp(-x))

def swish(x, beta):
    return x * sigmoid(beta * x)

x = np.arange(-6.0, 6.0, 0.1)
ax = get_central_ax()

legends = []
for beta in [1]:
    y_s = swish(x, beta)
    ax.plot(x, y_s, linestyle='-',color='r',linewidth=3)
    legends.append('β = '+str(beta))

ax.legend(legends)

plt.savefig("/Users/devil/Desktop/temp.png")
plt.show()