文章作者：Tyan
博客：noahsnail.com ?|? CSDN ?|? 簡(jiǎn)書(shū)
本文主要使用matplotlib進(jìn)行等高線圖的繪制。
- Demo
import matplotlib.pyplot as plt
import numpy as np
# 定義等高線高度函數(shù)
def f(x, y):
return (1 - x / 2 + x ** 5 + y ** 3) * np.exp(- x ** 2 - y ** 2)
# 數(shù)據(jù)數(shù)目
n = 256
# 定義x, y
x = np.linspace(-3, 3, n)
y = np.linspace(-3, 3, n)
# 生成網(wǎng)格數(shù)據(jù)
X, Y = np.meshgrid(x, y)
# 填充等高線的顏色, 8是等高線分為幾部分
plt.contourf(X, Y, f(X, Y), 8, alpha = 0.75, cmap = plt.cm.hot)
# 繪制等高線
C = plt.contour(X, Y, f(X, Y), 8, colors = 'black', linewidth = 0.5)
# 繪制等高線數(shù)據(jù)
plt.clabel(C, inline = True, fontsize = 10)
# 去除坐標(biāo)軸
plt.xticks(())
plt.yticks(())
plt.show()
- 結(jié)果
等高線圖
