~jrjohansson/qutip/master

« back to all changes in this revision

Viewing changes to test_qfunc.py

• browse files at revision 1
• compare with another revision
• download diff
• download tarball
• view history from revision 1

Show diffs side-by-side

added

removed

test_qfunc.py

 

1

#

 

2

#

 

3

#

 

4

#

 

5

from scipy import *

 

6

from scipy.linalg import *

 

7

import scipy.sparse as sp

 

8

import scipy.linalg as la

 

9

 

 

10

 

 

11

from qutip import *

 

12

 

 

13

from mpl_toolkits.mplot3d import Axes3D

 

14

from matplotlib import cm

 

15

from pylab import *

 

16

 

 

17

N = 20;

 

18

g = 2;

 

19

 

 

20

psi = basis(N,10) + basis(N,5)

 

21

 

 

22

print "psi = ", psi

 

23

 

 

24

xvec = arange(-40.,40.)*5./40

 

25

yvec = xvec

 

26

X,Y = meshgrid(xvec, yvec)

 

27

 

 

28

#W=wigner(psi,xvec,xvec)

 

29

#print "W = ", W

 

30

 

 

31

Q = qfunc(psi,xvec,xvec,g);

 

32

print "Q = ", Q

 

33

 

 

34

fig1 = plt.figure()

 

35

ax = Axes3D(fig1)

 

36

ax.plot_surface(X, Y, Q, rstride=2, cstride=2, cmap=cm.jet, alpha=0.7)

 

37

#ax.contour(X, Y, Q, levels=15, zdir='z', offset=-0.6)

 

38

#ax.set_zlim3d(-0.4,0.2)

 

39

#ax.set_zlim3d(0.,0.)

 

40

 

 

41

plt.show()

 

42

 

• Older »

Loggerhead is a web-based interface for Breezy
Version: 2.0.1