欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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

【运行结果】

使用 NumPy 和 Matplotlib 绘制函数图

如果需要函数图像以圆点呈现,而不是用线呈现,则需要将

plt.plot(x,y)改为plt.plot(x,y,'ob')

【运行结果】

使用 NumPy 和 Matplotlib 绘制函数图

正弦波 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()

【运行结果】

使用 NumPy 和 Matplotlib 绘制函数图

余弦函数只需将
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()

【运行结果】

使用 NumPy 和 Matplotlib 绘制函数图

普通图

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

【运行结果】

使用 NumPy 和 Matplotlib 绘制函数图

条形图

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

【运行结果】

使用 NumPy 和 Matplotlib 绘制函数图

散点图

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的散点图

使用 NumPy 和 Matplotlib 绘制函数图

等高线图

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

【运行结果】

使用 NumPy 和 Matplotlib 绘制函数图

灰度图

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

【运行结果】

使用 NumPy 和 Matplotlib 绘制函数图

饼状图

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

【运行结果】

使用 NumPy 和 Matplotlib 绘制函数图

量场图

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

【运行结果】

使用 NumPy 和 Matplotlib 绘制函数图

极轴图

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

【运行结果】

使用 NumPy 和 Matplotlib 绘制函数图

3d图

到此这篇关于使用 numpy 和 matplotlib 绘制函数图的文章就介绍到这了,更多相关numpy matplotlib 内容请搜索以前的文章或继续浏览下面的相关文章希望大家以后多多支持!