使用Matplotlib 3D軸，如何一次拖動兩個軸

Question

基本上我在一個圖中有兩個3D軸，一個是通過matplotlib.aninmation和一個3d線條並排構建的動畫。我想添加功能，以便當您旋轉一個軸時，另一個旋轉;例如，將當前視角發送到另一個軸的功能;

angle1 = getviewingangle(ax1) 
ax2.view_init(angle1) 
angle2 = getviewngangle(ax2) 
ax1.view_init(angle2) 

等。這是用於比較一個粒子的動畫路徑與它的預繪製軌跡。

Answer 1

爲了同步mplot3d中兩個子圖的旋轉，您可以將motion_notify_event連接到一個函數，該函數從旋轉圖中讀取角度並將其應用於相應的其他圖。

以下是添加了所述功能的圖庫示例。

from mpl_toolkits.mplot3d import Axes3D 
from matplotlib import cm 
import matplotlib.pyplot as plt 
import numpy as np 

n_angles = 36 
n_radii = 8 

radii = np.linspace(0.125, 1.0, n_radii) 
angles = np.linspace(0, 2*np.pi, n_angles, endpoint=False) 
angles = np.repeat(angles[..., np.newaxis], n_radii, axis=1) 

x = np.append(0, (radii*np.cos(angles)).flatten()) 
y = np.append(0, (radii*np.sin(angles)).flatten()) 
z = np.sin(-x*y) 

fig = plt.figure(figsize=(13,6)) 
fig.subplots_adjust(left=0, right=1, top=1, bottom=0, wspace=0) 
ax = fig.add_subplot(1, 2, 1, projection='3d') 
ax2 = fig.add_subplot(1, 2, 2, projection='3d') 

ax.plot_trisurf(x, y, z, cmap=cm.jet, linewidth=0.2) 
ax2.plot_trisurf(x, y, z, cmap=cm.viridis, linewidth=0.5) 

def on_move(event): 
    if event.inaxes == ax: 
     ax2.view_init(elev=ax.elev, azim=ax.azim) 
    elif event.inaxes == ax2: 
     ax.view_init(elev=ax2.elev, azim=ax2.azim) 
    else: 
     return 
    fig.canvas.draw_idle() 

c1 = fig.canvas.mpl_connect('motion_notify_event', on_move) 


plt.show() 

---

它可能是有意義的額外同步變焦實用爲好。在這種情況下，可以使用以下功能

def on_move(event): 
    if event.inaxes == ax: 
     if ax.button_pressed in ax._rotate_btn: 
      ax2.view_init(elev=ax.elev, azim=ax.azim) 
     elif ax.button_pressed in ax._zoom_btn: 
      ax2.set_xlim3d(ax.get_xlim3d()) 
      ax2.set_ylim3d(ax.get_ylim3d()) 
      ax2.set_zlim3d(ax.get_zlim3d()) 
    elif event.inaxes == ax2: 
     if ax2.button_pressed in ax2._rotate_btn: 
      ax.view_init(elev=ax2.elev, azim=ax2.azim) 
     elif ax2.button_pressed in ax2._zoom_btn: 
      ax.set_xlim3d(ax2.get_xlim3d()) 
      ax.set_ylim3d(ax2.get_ylim3d()) 
      ax.set_zlim3d(ax2.get_zlim3d()) 
    else: 
     return 
    fig.canvas.draw_idle()