使用 NumPy 和 Matplotlib 绘制函数图
程序员文章站
2022-03-28 20:09:11
matplotlib是python的绘图库,它与numpy一起使用【引例】画出y=2*x+5的函数图像import numpy as npfrom matplotlib import pyplot a...
matplotlib是python的绘图库,它与numpy一起使用
【引例】
画出y=2*x+5的函数图像
import numpy as np from matplotlib import pyplot as plt #设置横坐标取值、确定函数表达式 x = np.arange(1, 11) y = 2 * x + 5 #添加标题、横坐标、纵坐标标签 plt.title("matplotlib demo") plt.xlabel("x axis caption") plt.ylabel("y axis caption") #画函数图像 plt.plot(x, y) plt.show()
【运行结果】
如果需要函数图像以圆点呈现,而不是用线呈现,则需要将 plt.plot(x,y)改为plt.plot(x,y,'ob')
【运行结果】
正弦波 y=sin(x)
import numpy as np from matplotlib import pyplot as plt #设置横坐标取值、确定函数表达式 x = np.arange(0, 3*np.pi,0.1) y = np.sin(x) #添加标题、横坐标、纵坐标标签 plt.title("y=sin(x)") plt.xlabel("x axis caption") plt.ylabel("y axis caption") #画函数图像 plt.plot(x, y) plt.show()
【运行结果】
余弦函数只需将 y = np.sin(x) 改为 y = np.cos(x)即可
subplot()函数
应用subplot()函数,在同一图中绘制不同的东西
# relu和sigmoid激活函数示意图 import numpy as np import matplotlib.pyplot as plt import matplotlib.patches as patches #设置图片大小 plt.figure(figsize=(6, 4)) # x是1维数组,数组大小是从-10. 到10.的实数,每隔0.1取一个点 x = np.arange(-10, 10, 0.1) # 计算 sigmoid函数 s = (np.exp(x)-np.exp(-x))/(np.exp(x)+np.exp(-x)) # 计算relu函数 y = np.clip(x, a_min = 0., a_max = none) ######################################################### # 以下部分为画图程序 # 设置两个子图窗口,将sigmoid的函数图像画在上边 f = plt.subplot(2,1,1) # 画出函数曲线 plt.plot(x, s, color='r') # 添加文字说明 plt.text(-5., 0.9, r'$y=sigmoid(x)$', fontsize=13) # 设置坐标轴格式 currentaxis=plt.gca() currentaxis.xaxis.set_label_text('x', fontsize=15) currentaxis.yaxis.set_label_text('y', fontsize=15) # 将relu的函数图像画在下边 f = plt.subplot(2,1,2) # 画出函数曲线 plt.plot(x, y, color='g') # 添加文字说明 plt.text(-3.0, 9, r'$y=relu(x)$', fontsize=13) # 设置坐标轴格式 currentaxis=plt.gca() currentaxis.xaxis.set_label_text('x', fontsize=15) currentaxis.yaxis.set_label_text('y', fontsize=15) plt.show()
【运行结果】
普通图
from matplotlib import pyplot as plt import numpy as np n = 256 #n可以理解为曲线的平滑程度 x = np.linspace(-np.pi,np.pi,n,endpoint=true) y = np.sin(2*x) plt.plot(x,y+1,color='red',alpha=1.00) plt.plot(x,y-1,color='blue',alpha=1.00) plt.show()
【运行结果】
条形图
from matplotlib import pyplot as plt #数据组1 x = [5,8,11] #横坐标 y = [12,16,6] #对应值 #数据组2 x2 = [6,9,12] #横坐标 y2 = [6,15,7] #对应值 plt.bar(x, y, color = 'r', align = 'center') plt.bar(x2, y2, color = 'g', align = 'center') plt.title('bar graph') plt.ylabel('y axis') plt.xlabel('x axis') plt.show()
【运行结果】
散点图
from matplotlib import pyplot as plt import numpy as np n = 1024 x = np.random.normal(0,30,n) y = np.random.normal(0,30,n) plt.title('scale:30') plt.scatter(x,y,color='red') plt.show()
【运行结果】以下分别是方差分别为5,10,15,20,25,30的散点图
等高线图
from matplotlib import pyplot as plt import numpy as np def f(x,y):return(1-x/2+x**5+y**3)*np.exp(-x**2-y**2) #等高线模型 n = 256 #n可以理解为曲线的平滑程度 x = np.linspace(-3,3,n) y = np.linspace(-3,3,n) x,y = np.meshgrid(x,y) plt.contourf(x,y,f(x,y),8,alpha=.75,cmap='jet') c = plt.contour(x,y,f(x,y),8,color='black',linewidth=.5) plt.show()
【运行结果】
灰度图
from matplotlib import pyplot as plt import numpy as np def f(x,y):return(1-x/2+x**5+y**3)*np.exp(-x**2-y**2) n = 10 x = np.linspace(-3,3,4*n) y = np.linspace(-3,3,4*n) x,y = np.meshgrid(x,y) plt.imshow(f(x,y)) plt.show()
【运行结果】
饼状图
from matplotlib import pyplot as plt import numpy as np n = 20 #n决定了有多少个分区,该代码显示有20个分区 z = np.random.uniform(0,1,n) plt.pie(z) plt.show()
【运行结果】
量场图
from matplotlib import pyplot as plt import numpy as np n = 8 x,y = np.mgrid[0:n,0:n] plt.quiver(x,y,color="red") plt.show()
【运行结果】
极轴图
from matplotlib import pyplot as plt import numpy as np plt.figure(figsize=(4,4)) ax1 = plt.subplot(111,projection='polar') ax1.set_title('spot fish') ax1.set_rlim(0,12) data = np.random.randint(1,10,10) theta = np.arange(0,2*np.pi,2*np.pi/10) bar = ax1.bar(theta,data,alpha=0.5) for r,bar in zip(data,bar): bar.set_facecolor(plt.cm.jet(r/10.)) plt.show()
【运行结果】
3d图
到此这篇关于使用 numpy 和 matplotlib 绘制函数图的文章就介绍到这了,更多相关numpy matplotlib 内容请搜索以前的文章或继续浏览下面的相关文章希望大家以后多多支持!
推荐阅读
-
Python学习笔记--使用matplotlib绘制饼状图
-
使用matplotlib库绘制函数图
-
使用Revit和CAD绘制出螺旋型屋面效果图的方法
-
python使用matplotlib在一个图形中绘制多个子图以及一个子图中绘制多条动态折线问题
-
在Flash CS6中使用with函数绘制背景图教程
-
python使用matplotlib绘制热图
-
Python使用matplotlib的pie函数绘制饼状图功能示例
-
Python3使用Matplotlib 绘制精美的数学函数图形
-
在Linux下使用Python的matplotlib绘制数据图的教程
-
matplotlib 画动态图以及plt.ion()和plt.ioff()的使用详解