基于matplotlib绘制立体爱心

        由于到了520这种特殊的日子，用什么来表达爱意，就是个问题。本文根据通过matplotlib绘制心形 ，教大家绘制一个立体爱心。

        本文在源代码的基础上，进行优化。使用plt.axis('off')和ax.view_init(elev=5., azim=-76)两条语句，分别取消坐标轴和更改最初视角。

import bbox as bbox
from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm
from matplotlib.ticker import LinearLocator, FormatStrFormatter
import matplotlib.pyplot as plt
import numpy as np



def heart_3d(x, y, z):
    return (x ** 2 + (9 / 4) * y ** 2 + z ** 2 - 1) ** 3 - x ** 2 * z ** 3 - (9 / 80) * y ** 2 * z ** 3


def heart_3d_2(x, y, z):
    return (2 * x ** 2 + 2 * y ** 2 + z ** 2 - 1) ** 3 - 0.1 * x ** 2 * z ** 3 - y ** 2 * z ** 3


def plot_implicit(fn, bbox=(-1.5, 1.5)):
    
    xmin, xmax, ymin, ymax, zmin, zmax = bbox * 3
    fig = plt.figure()
    ax = fig.add_subplot(111, projection='3d')
    A = np.linspace(xmin, xmax, 100)  
    B = np.linspace(xmin, xmax, 40)  
    A1, A2 = np.meshgrid(A, A)  

    for z in B:  
        X, Y = A1, A2
        Z = fn(X, Y, z)
        cset = ax.contour(X, Y, Z + z, [z], zdir='z', colors=('r',))
       

    for y in B:  
        X, Z = A1, A2
        Y = fn(X, y, Z)
        cset = ax.contour(X, Y + y, Z, [y], zdir='y', colors=('red',))

    for x in B:  
        Y, Z = A1, A2
        X = fn(x, Y, Z)
        cset = ax.contour(X + x, Y, Z, [x], zdir='x', colors=('red',))

  
    ax.set_zlim3d(zmin, zmax)
    ax.set_xlim3d(xmin, xmax)
    ax.set_ylim3d(ymin, ymax)
    #改变视角
    ax.view_init(elev=5., azim=-76)
    # 取消坐标轴显示
    plt.axis('off')
    plt.show()


if __name__ == '__main__':
     plot_implicit(heart_3d)