我是Python新手,现在正在学习matplotlib
来使用动画函数。
因为我的学习,我遵循qutip
教程。
但是当我复制并粘贴qutip
教程的示例代码时,它没有起作用
错误消息是Axes3D object is not iterable
。
所以,我想检查我创建的代码,但不知道问题是不是我的代码是另一回事。
我不知道该怎么做,我想知道为什么示例代码不能工作。在
以下是教程中的示例代码:
from qutip import *
from scipy import *
def qubit_integrate(w, theta, gamma1, gamma2, psi0, tlist):
sx = sigmax(); sy = sigmay(); sz = sigmaz(); sm = sigmam()
H = w * (cos(theta) * sz + sin(theta) * sx)
c_op_list = []
n_th = 0.5 # temperature
rate = gamma1 * (n_th + 1)
if rate > 0.0: c_op_list.append(sqrt(rate) * sm)
rate = gamma1 * n_th
if rate > 0.0: c_op_list.append(sqrt(rate) * sm.dag())
rate = gamma2
if rate > 0.0: c_op_list.append(sqrt(rate) * sz)
output = mesolve(H, psi0, tlist, c_op_list, [sx, sy, sz])
return output.expect[0], output.expect[1], output.expect[2]
w = 1.0 * 2 * pi # qubit angular frequency
theta = 0.2 * pi # qubit angle from sigma_z axis (toward sigma_x axis)
gamma1 = 0.5 # qubit relaxation rate
gamma2 = 0.2 # qubit dephasing rate
a = 1.0
psi0 = (a* basis(2,0) + (1-a)*basis(2,1))/(sqrt(a**2 + (1-a)**2))
tlist = linspace(0,4,250)
sx, sy, sz = qubit_integrate(w, theta, gamma1, gamma2, psi0, tlist)
from pylab import *
import matplotlib.animation as animation
from mpl_toolkits.mplot3d import Axes3D
fig = figure()
ax = Axes3D(fig,azim=-40,elev=30)
sphere = Bloch(axes=ax)
def animate(i):
sphere.clear()
sphere.add_vectors([np.sin(theta),0,np.cos(theta)])
sphere.add_points([sx[:i+1],sy[:i+1],sz[:i+1]])
sphere.make_sphere()
return ax
def init():
sphere.vector_color = ['r']
return ax
ani = animation.FuncAnimation(fig, animate, np.arange(len(sx)),
init_func=init, blit=True, repeat=False)
ani.save('bloch_sphere.mp4', fps=20, clear_temp=True)
这是我自己的代码:
^{pr2}$
修复教程
删除
blit
参数使教程正常工作:修正你的例子
frames
必须是iterable。在更改:
^{pr2}$进入:
也就是说,这一行:
应该变成这样:
再做一些改变。在
state(i)
而不是方括号state[i]
ani = ani.FuncAnimation
完整代码:
这是动画的结束状态:
相关问题 更多 >
编程相关推荐