AnalysisFunctions.py 文件源码

def create_kgrid(nx, ny, nz, lx=2*pi, ly=2*pi, lz=2*pi):
   """
   Create a 3D k grid for Fourier space calculations
   """

   print lx, ly, lz

   kx = nf.fftshift(nf.fftfreq(nx))*nx*2*pi/lx
   ky = nf.fftshift(nf.fftfreq(ny))*ny*2*pi/ly
   kz = nf.fftshift(nf.fftfreq(nz))*nz*2*pi/lz

   mg = np.meshgrid(kx,ky,kz)

   km = np.sqrt(np.sum((m**2 for m in mg)))

   return kx[:,nna,nna], ky[nna,:,nna], kz[nna,nna,:], km

#==================================================