如何从 matplotlib 图中获取方位角和仰角

Question

我正在使用 matplotlib.pyplot 和 numpy 绘制一组 3D 点。使用鼠标旋转图像会在图形窗口底部生成方位角和仰角。

如何以编程方式获取这些值，以便在后续计算中使用它们，例如当释放鼠标或单击屏幕上的按钮时。

我正在使用 python 2.7.5。我下面的代码应该能说明问题。

import numpy as np

import matplotlib.pyplot as plt

from mpl_toolkits.mplot3d import Axes3D

mol=np.random.rand(6,3)*10.0

fig3d = plt.figure(figsize = ( 6.5, 6.5 ) )

fig3d.canvas.set_window_title('3D')

ax3d = fig3d.gca(  projection='3d')

ax3d.scatter( mol[:,0], mol[:,1], mol[:,2], s=200)  

plt.show()

Answer 1

下面是一些代码，它使用回调打印出当前的方位角和仰角，并将它们附加到列表中以供以后进一步使用。这些详细信息分别保存在ax3d.azim 和ax3d.elev 中。

每次您释放鼠标按钮时，它都会调用on_click。这将允许您简单地将图形旋转到您选择的旋转，然后保存投影。

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

mol=np.random.rand(6,3)*10.0

fig3d = plt.figure(figsize = ( 6.5, 6.5 ) )

fig3d.canvas.set_window_title('3D')

ax3d = fig3d.gca(projection='3d')
ax3d.scatter(mol[:,0], mol[:,1], mol[:,2], s=200)  

# List to save your projections to
projections = []

# This is called everytime you release the mouse button
def on_click(event):
    azim, elev = ax3d.azim, ax3d.elev
    projections.append((azim, elev))
    print(azim, elev)

cid = fig3d.canvas.mpl_connect('button_release_event', on_click)

plt.show()