from qutip import *import numpy as npimport matplotlib.pyplot as pltfrom qutip import gatesimport mathfrom time import clockpi=np.piwd = 5.0 * 2 * pi # drive frequencywa = 5.0 * 2 * pi # atom frequencyomega = 0.05 * 2 * pi # couplingtlist = np.linspace(0,5000,100001)psi0 = (basis(2,0)+ basis(2,1)).unit() # start with an excited atom#psi0=psi0.ptrace(0)print(psi0)sm = destroy(2)H0 = wa*sigmaz()/2output2 = mesolve(H0, psi0, tlist, [], [])plt.figure()n_q=[]subplot(3,1,1)#for t in range(0,len(tlist)):# U=(np.exp(1j*wd*tlist[t])*sm.dag()*sm+sm*sm.dag())#.dag()# op=U*sigmax()*U.dag()# n_q.append(expect(op,output2.states[t]))n_q=expect(sm.dag()+sm,output2.states)plot(tlist,n_q,label='sigma_x')title(psi0)n_q=[]subplot(3,1,2)#for t in range(0,len(tlist)):# U=(np.exp(1j*wd*tlist[t])*sm.dag()*sm+sm*sm.dag())#.dag()# op=U*sigmay()*U.dag()# n_q.append(expect(op,output2.states[t]))n_q=expect(1j*(sm.dag()-sm),output2.states)plot(tlist,n_q,label='sigma_y')
subplot(3,1,3)plot(tlist,expect(sm.dag()*sm,output2.states),label='sigma_z')
Initial state is a density matrix: