三次元プロット
ぐるぐる回す
アニメーション

from IPython.display import HTML,display_html
import numpy as np
from matplotlib import pyplot as plt
import matplotlib.animation as anime

def plot3d(
    fname,func,itrot,itshw,itfrz,
    xrng=(-1,1),
    yrng=(-1,1),
    zrng=(-1,1),
    nx=100,
    ny=100,
    fpfrz=lambda i:0,
    fps=10,
    dpi=100,
    color="#FF0000",
    lw=0.3,
    elv=20,
    azm=30
    ):
    fig,ax=plt.subplots(figsize=(6,6),subplot_kw={'projection':'3d'})
    def plot(i):
        ax.cla()
        if (i>=itrot+itshw):
            x=np.linspace(xrng[0],xrng[1],nx)
            y=np.linspace(yrng[0],yrng[1],ny)
            X,Y=np.meshgrid(x,y)
            ax.plot_wireframe(X,Y,func(X,Y),color=color,linewidth=lw)
            fpfrz(i-itrot-itshw,ax)
            ax.view_init(elev=elv,azim=azm)
            ax.set_xlim(xrng[0],xrng[1])
            ax.set_ylim(yrng[0],yrng[1])
            ax.set_zlim(zrng[0],zrng[1])
        elif (i>=itrot):
            x=np.linspace(xrng[0],xrng[1],nx)
            y=np.linspace(yrng[0],yrng[1],ny)
            X,Y=np.meshgrid(x,y)
            ax.plot_wireframe(X,Y,func(X,Y),color=color,linewidth=lw)
            ax.set_xlim(xrng[0],xrng[1])
            ax.set_ylim(yrng[0],yrng[1])
            ax.set_zlim(zrng[0],zrng[1])
            ax.view_init(elev=elv,azim=azm,)
        else:
            x=np.linspace(xrng[0],xrng[1],nx)
            y=np.linspace(yrng[0],yrng[1],ny)
            X,Y=np.meshgrid(x,y)
            ax.plot_wireframe(X,Y,func(X,Y),color=color,linewidth=lw)
            ax.set_xlim(xrng[0],xrng[1])
            ax.set_ylim(yrng[0],yrng[1])
            ax.set_zlim(zrng[0],zrng[1])
            ax.view_init(elev=elv,azim=(360/itrot)*i+azm,)
    anim=anime.FuncAnimation(fig,plot,frames=itrot+itshw+itfrz)
    anim.save(fname,fps=fps,dpi=dpi)

    plt.close(fig)
    display_html(HTML(anim.to_jshtml(fps=fps)))

def plot3d_expand(
    fname,func,itrot,itshw,itexp,itfrz,
    xrng0=(-1,1),
    yrng0=(-1,1),
    zrng0=(-1,1),
    xrng1=(-1,1),
    yrng1=(-1,1),
    zrng1=(-1,1),
    nx=100,
    ny=100,
    fpfrz=lambda i:0,
    fps=10,
    dpi=100,
    color="#FF0000",
    ucolor="#7F7F7F",
    lw=0.3,
    elv=20,
    azm=30
    ):
    fig,ax=plt.subplots(figsize=(6,6),subplot_kw={'projection':'3d'})
    def plot(i):
        ax.cla()
        if (i>=itrot+itexp+itshw*2):
            x=np.linspace(xrng1[0],xrng1[1],nx)
            y=np.linspace(yrng1[0],yrng1[1],ny)
            X,Y=np.meshgrid(x,y)
            ax.plot_wireframe(X,Y,func(X,Y),color=color,linewidth=lw)
            fpfrz(i-itrot-itexp-itshw*2,ax)
            ax.view_init(elev=elv,azim=azm)
            ax.set_xlim(xrng1[0],xrng1[1])
            ax.set_ylim(yrng1[0],yrng1[1])
            ax.set_zlim(zrng1[0],zrng1[1])
        if (i>=itrot+itexp+itshw):
            x=np.linspace(xrng1[0],xrng1[1],nx)
            y=np.linspace(yrng1[0],yrng1[1],ny)
            X,Y=np.meshgrid(x,y)
            ax.plot_wireframe(X,Y,func(X,Y),color=color,linewidth=lw)
            ax.view_init(elev=elv,azim=azm)
            ax.set_xlim(xrng1[0],xrng1[1])
            ax.set_ylim(yrng1[0],yrng1[1])
            ax.set_zlim(zrng1[0],zrng1[1])
        elif (i>=itrot+itshw):
            p=((itrot+itexp+itshw-i)/itexp)
            x0,x1=xrng0[0]*p+(1-p)*xrng1[0],xrng0[1]*p+(1-p)*xrng1[1]
            y0,y1=yrng0[0]*p+(1-p)*yrng1[0],yrng0[1]*p+(1-p)*yrng1[1]
            z0,z1=zrng0[0]*p+(1-p)*zrng1[0],zrng0[1]*p+(1-p)*zrng1[1]
            x=np.linspace(x0,x1,nx)
            y=np.linspace(y0,y1,ny)
            X,Y=np.meshgrid(x,y)
            ax.plot_wireframe(X,Y,func(X,Y),color=ucolor,linewidth=lw)
            x=np.linspace(xrng1[0],xrng1[1],nx)
            y=np.linspace(yrng1[0],yrng1[1],ny)
            X,Y=np.meshgrid(x,y)
            ax.plot_wireframe(X,Y,func(X,Y),color=color,linewidth=lw)
            ax.view_init(elev=elv,azim=azm)
            ax.set_xlim(x0,x1)
            ax.set_ylim(y0,y1)
            ax.set_zlim(z0,z1)
        elif (i>=itrot):
            x=np.linspace(xrng0[0],xrng0[1],nx)
            y=np.linspace(yrng0[0],yrng0[1],ny)
            X,Y=np.meshgrid(x,y)
            ax.plot_wireframe(X,Y,func(X,Y),color=ucolor,linewidth=lw)
            x=np.linspace(xrng1[0],xrng1[1],nx)
            y=np.linspace(yrng1[0],yrng1[1],ny)
            X,Y=np.meshgrid(x,y)
            ax.plot_wireframe(X,Y,func(X,Y),color=color,linewidth=lw)
            ax.set_xlim(xrng0[0],xrng0[1])
            ax.set_ylim(yrng0[0],yrng0[1])
            ax.set_zlim(zrng0[0],zrng0[1])
            ax.view_init(elev=elv,azim=azm,)
        else:
            x=np.linspace(xrng0[0],xrng0[1],nx)
            y=np.linspace(yrng0[0],yrng0[1],ny)
            X,Y=np.meshgrid(x,y)
            ax.plot_wireframe(X,Y,func(X,Y),color=ucolor,linewidth=lw)
            ax.set_xlim(xrng0[0],xrng0[1])
            ax.set_ylim(yrng0[0],yrng0[1])
            ax.set_zlim(zrng0[0],zrng0[1])
            ax.view_init(elev=elv,azim=(360/itrot)*i+azm,)
    anim=anime.FuncAnimation(fig,plot,frames=itrot+itshw*2+itexp+itfrz)
    anim.save(fname,fps=fps,dpi=dpi)

    plt.close(fig)
    display_html(HTML(anim.to_jshtml(fps=fps)))

func=lambda x,y:np.exp(-(x/0.5)**2-(y/0.3)**2)*np.sin(x*7)*np.cos(y*10)
def plot(i,ax):
    x,y=np.random.uniform(0,0.6),np.random.uniform(-0.2,0.4)
    ax.scatter(x,y,func(x,y))
plot3d("a.gif",func,36,5,20,(-1,1),(-1,1),(-1,1),200,200,fpfrz=plot)
plot3d_expand("b.gif",func,36,5,14,20,(-1,1),(-1,1),(-1,1),(0,0.6),(-0.2,0.4),(-0.4,1),200,200,fpfrz=plot)