今天,当我使用python中的开源量子工具箱QuTiP绘制相干态的Wigner函数时,我注意到了一些奇怪的事情。
当我进行绘图时,我注意到在绘图边缘附近不应该存在的这些奇怪的图案。我相信这只是某种数字错误,但我不知道如何摆脱或最小化它们或最公正的东西:是什么原因造成的。
这是代码
# import packages
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.colors as colors
import matplotlib as mpl
from matplotlib import cm
from qutip import *
N = 60 # number of levels in Hilbert space
# density matrix of a coherent state
rho_coherent = coherent_dm(N, 1-1j)
X = np.linspace(-3, 3, 300)
Y = np.linspace(-3, 3, 300)
# Wigner function
W = wigner(rho_coherent, X, Y, 'iterative', 2)
X, Y = np.meshgrid(X, Y)
# Color Normalization
class MidpointNormalize(colors.Normalize):
def __init__(self, vmin=None, vmax=None, midpoint=None, clip=False):
self.midpoint = midpoint
colors.Normalize.__init__(self, vmin, vmax, clip)
def __call__(self, value, clip=None):
x, y = [self.vmin, self.midpoint, self.vmax], [0, 0.5, 1]
return np.ma.masked_array(np.interp(value, x, y))
# contour plot
plt.subplot(111, aspect='equal')
plt.contourf(X, Y, W, 100, cmap = cm.RdBu_r, norm = MidpointNormalize(midpoint=0.))
plt.show()
这是情节
您应该清楚地看到边缘周围的蓝色斑点不应该存在!蓝色斑点表示此时的维格纳函数为负,但相干状态的维格纳函数在各处都为正!
我还注意到,当我将linspace步骤从300减少到100时,蓝色部分消失了。
如果有人可以解释是什么导致了此问题的出现,我们将不胜感激。
答案 0 :(得分:2)
这仅仅是由于截断。当使用有限数量的模式(在您的情况下为N = 60)时,维格纳函数会在某些时候变为负值。
减少linspace步距会使图中显示的负区域变为零值增量,并将这些区域显示为零。减少linspace步骤可能是解决您问题的最佳解决方案。您的绘图将仅与截断所引入的错误一样准确,因此只需降低分辨率,直到这些错误消失即可。