Matplotlib学习笔记（五）

最新推荐文章于 2021-04-28 13:47:10 发布

DTree123

最新推荐文章于 2021-04-28 13:47:10 发布

阅读量414

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.csdn.net/MainTain_/article/details/77185892

Python 专栏收录该内容

24 篇文章

订阅专栏

本文介绍如何使用Python中的Matplotlib库绘制等高线图和3D图。通过具体的代码示例展示了如何创建数据网格、设置颜色映射以及添加轮廓线等操作。

条形图：

结果：

********************************************************************************************************************************************

画等高线图：

import matplotlib.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
x = np.linspace(-3,3,n)
y = np.linspace(-3,3,n)
X,Y = np.meshgrid(x,y)

#use plt.contourf to filling contours
#X,Y and value for (X,Y)point
plt.contourf(X,Y,f(X,Y),8,alpha=0.75,cmap=plt.cm.hot)

#use plt.contour to add contour lines
C = plt.contour(X,Y,f(X,Y),8,colors='black',linewidth=.5)

#adding label
plt.clabel(C,inline=True,fontsize=10)

plt.xticks(())
plt.yticks(())
plt.show()

如图：

*******************************************************************************************************************

画图：

结果：

****************************************************************************************************************************************

画3D图：

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

fig = plt.figure()
ax = Axes3D(fig)
X = np.arange(-4,4,0.25)
Y = np.arange(-4,4,0.25)
X,Y = np.meshgrid(X,Y)
R = np.sqrt(X**2+Y**2)
#height value
Z = np.sin(R)

ax.plot_surface(X,Y,Z,rstride=1,cstride=1,cmap=plt.get_cmap('rainbow'))

ax.contour(X,Y,Z,zdir='z',offset=-2,cmap='rainbow')

ax.set_zlim(-2,2)
plt.show()

如下：