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()

如何从matplotlib图中获得方位角和高程

1 个答案: