2017-02-13 307 views
1

我有一個使用matplotlib繪製的三維圖形。三維matplotlib圖形中圖像投影的控制角度

我想有在特定的角度投射3D圖,但使用 axis.view_init(elevation_angle,azimuthal_angle)並不包括我想要的角度,這是由於在3旋轉的3架可能的飛機尺寸和只有2個角度,你可以在matplotlib中指定。

這是一個最小工作示例。

import numpy,matplotlib,scipy 
from matplotlib import pyplot 
from scipy import constants 
from mpl_toolkits.mplot3d.axes3d import Axes3D 

pi=scipy.constants.pi 

    w= numpy.arange(0,5000,1) 


def lorentzian(x,center,width,time,tau): 
    return 1.0/((1+((x-center)/width)**2)*width*pi)*numpy.exp(-time/tau) 

centers= [1000.0,2000.0,4000.0] 
widths = [100.0,300.0,50.0] 

taus=numpy.zeros(len(widths)) 

for i in xrange(0,len(widths),1): 
    taus[i] = 1.0/widths[i] 

time = numpy.array([0,0.019,0.033]) 
B = numpy.zeros((3,len(w))) 
for z in xrange(0,len(time),1): 
    a= numpy.zeros(len(w)) 
    for i in xrange(0,len(centers),1): 
     a = a + lorentzian(w,centers[i],widths[i],time[z],taus[i]) 

    B[z] = a 

ThreeD_spectrum_figure = matplotlib.pyplot.figure() 
ThreeD_spectrum_axis = ThreeD_spectrum_figure.add_subplot(111, projection="3d") 

colour = ["purple","green","orange"] 
for i in xrange(0,len(time),1): 
    ThreeD_spectrum_axis.plot(w, B[i],time[i],color=colour[i],linewidth=3) 

ThreeD_spectrum_axis.view_init(135,-90) 
matplotlib.pyplot.show() 

這將生成下圖中顯示的圖形,我有,在ms paint中快速繪製,添加的線條顯示旋轉和軸。

紅色顯示繞Y軸旋轉的方位角控制。

藍色顯示圍繞X軸旋轉的仰角。

黑色顯示我需要的旋轉,它將圍繞Z軸旋轉,這是使用axis.view_init(elevation_angle,azimuth_angle)不可能實現的。 enter image description here

任何人都可以提供任何幫助或見解我如何能夠實現這種輪換? 謝謝

回答

1

圍繞z軸的旋轉可以單獨使用方位角來實現。

enter image description here

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

pi=np.pi 
w= np.arange(0,5000,1) 

def lorentzian(x,center,width,time,tau): 
    return 1.0/((1+((x-center)/width)**2)*width*pi)*np.exp(-time/tau) 

centers= [1000.0,2000.0,4000.0] 
widths = [100.0,300.0,50.0] 

taus=np.zeros(len(widths)) 

for i in xrange(0,len(widths),1): 
    taus[i] = 1.0/widths[i] 

time = np.array([0,0.019,0.033]) 
B = np.zeros((3,len(w))) 
for z in xrange(0,len(time),1): 
    a= np.zeros(len(w)) 
    for i in xrange(0,len(centers),1): 
     a = a + lorentzian(w,centers[i],widths[i],time[z],taus[i]) 
    B[z] = a 

fig = plt.figure() 
ax = fig.add_subplot(111, projection="3d") 

colour = ["purple","green","orange"] 
for i in xrange(len(time)-1,-1,-1): 
    ax.plot(time[i]*np.ones_like(w), w ,B[i]*np.ones_like(w),color=colour[i],linewidth=3) 

ax.set_xlabel("x") 
ax.set_ylabel("y") 
ax.set_zlabel("z") 

#(elev, azim) 
ax.view_init(45,0) 

phi = np.linspace(0, 2*np.pi) 

def update(phi): 
    ax.view_init(45, phi*180./np.pi) 

ani = matplotlib.animation.FuncAnimation(fig, update, frames=phi) 
ani.save(__file__+".gif", writer='imagemagick', fps=10) 
plt.show()