三维函数的可视化 密度图和等高线图

用等高线图可视化一个三维函数，代码如下：

import numpy as np
from matplotlib import pyplot as plt

plt.style.use('seaborn-white')
%matplotlib inline

def f(x, y):
    z = np.sin(x) ** 9 + np.cos(13 + x * y) * np.cos(x)
    return z

x = np.linspace(0, 5, 70)
y = np.linspace(0, 5, 60)
X,Y = np.meshgrid(x, y)
Z = f(X, Y)

plt.figure(figsize=(9,6))
plt.contour(X, Y, Z, colors='black')

 

图形如下：

 

 

 用彩色等高线重画上面的图，代码如下:

plt.figure(figsize=(9,6))
plt.contour(X, Y, Z, 20, cmap='RdGy')

图形如下：