码迷,mamicode.com
首页 > 其他好文 > 详细

Contours 等高线图

时间:2018-07-29 23:27:18      阅读:218      评论:0      收藏:0      [点我收藏+]

标签:back   line   没有   inline   一个   label   图片   图像   绘制   

1、画等高线

数据集即三维点 (x,y) 和对应的高度值,共有256个点。高度值使用一个 height function f(x,y) 生成。 x, y 分别是在区间 [-3,3] 中均匀分布的256个值,并用meshgrid在二维平面中将每一个x和每一个y分别对应起来,编织成栅格:

import matplotlib.pyplot as plt
import numpy as np

def f(x,y):
    # the height function
    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)

接下来进行颜色填充。使用函数plt.contourf把颜色加进去,位置参数分别为:X, Y, f(X,Y)。透明度0.75,并将 f(X,Y) 的值对应到color map的暖色组中寻找对应颜色。

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

接下来进行等高线绘制。使用plt.contour函数划线。位置参数为:X, Y, f(X,Y)。颜色选黑色,线条宽度选0.5。现在的结果如下图所示,只有颜色和线条,还没有数值Label:

# use plt.contour to add contour lines
C = plt.contour(X, Y, f(X, Y), 8, colors=black, linewidth=.5)
#其中,8代表等高线的密集程度,这里被分为10个部分。如果是0,则图像被一分为二。

技术分享图片

2、添加高度数字

最后加入Label,inline控制是否将Label画在线里面,字体大小为10。并将坐标轴隐藏:

 

plt.clabel(C, inline=True, fontsize=10)
plt.xticks(())
plt.yticks(())

 

 

 

技术分享图片

Contours 等高线图

标签:back   line   没有   inline   一个   label   图片   图像   绘制   

原文地址:https://www.cnblogs.com/anhoo/p/9387968.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!