【数学】绘制函数图像

函数图像用来 验证作业的结果

import matplotlib.pyplot as pt
import numpy,math


#设置函数
def f(x):
    if x<0:
        return 2+x
    elif x==0:
        return 1
    elif x>0:
        return 3+ math.cos(x)
    else:
        return 0

#设置坐标轴范围
x_row = numpy.linspace(-20,20,500)
y_colomn = [f(x) for x in x_row]

pt.figure()

ax = pt.gca()
ax.spines['right'].set_color('white')
ax.spines['top'].set_color("white")

ax.xaxis.set_ticks_position('bottom')
ax.yaxis.set_ticks_position('left')

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

pt.plot(x_row,y_colomn)
pt.show()