[第一遍文章预皇,也算熟悉一下Markdown] 摘自:腳本之家技即,詳細(xì)原文請點(diǎn)此處
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
# 計(jì)算x,y坐標(biāo)對應(yīng)的高度值
def f(x, y):
return (1-x/2+x**5+y**3) * np.exp(-x**2-y**2)
# 生成x,y的數(shù)據(jù)
n = 256
x = np.linspace(-3, 3, n)
y = np.linspace(-3, 3, n)
#把x,y數(shù)據(jù)生成mesh網(wǎng)格狀的數(shù)據(jù),因?yàn)榈雀呔€的顯示是在網(wǎng)格的基礎(chǔ)上添加上高度值
X, Y = np.meshgrid(x, y)
#填充等高線
plt.contourf(X, Y, f(X, Y), 20, cmap=plt.cm.hot)
#添加等高線
C = plt.contour(X, Y, f(X, Y), 20)
plt.clabel(C, inline=True, fontsize=12)
#顯示圖表
plt.show()