for b in range(len(kzgrid)): for j in range(len(herm_grid) + 1): if j < (len(herm_grid) - 1): g_tp[v, w, b, j] = (g_t[v, w, b, j] + g_t[v, w, b, j + 1]) / 2 g_tm[v, w, b, j] = (g_t[v, w, b, j] - g_t[v, w, b, j + 1]) / 2 ################################## #Entropy entn_sum[:, 10] = entn_sum[:, 10] + get_entropy_hermite( gt0, kzind=-1, include_kz0=include_kz0) for k in range(10): kzindex = k * par['nkz0'] / 20 #print 'kzindex',kzindex entn_sum[:, k] = entn_sum[:, k] + dd.get_entropy_hermite(gt0, kzind=kzindex) entnp_sum[:, k] = entnp_sum[:, k] + dd.get_entropy_hermite(g_tp, kzind=kzindex) entnm_sum[:, k] = entnm_sum[:, k] + dd.get_entropy_hermite(g_tm, kzind=kzindex) entn_sum = entn_sum / float(ntime) entnp_sum = entnp_sum / float(ntime) entnm_sum = entnm_sum / float(ntime) plt.loglog(herm_grid, prefactor * entn_sum[:, 10], basex=10, basey=10) temp = prefactor * entn_sum[20, 10] temp = temp / (herm_grid**(-1))[20]
iend = np.argmin(abs(time - end_time)) ntime = iend - istart + 1 entn_sum = np.zeros((par['nv0'], 11), dtype='float') for i in range(istart, iend + 1): print 'time=', time[i], ' of ', time[iend] gt0 = dd.read_time_step_g(i) gt0 = np.reshape(gt0, (par['nkx0'], par['nky0'], par['nkz0'], par['nv0']), order='F') #Entropy entn_sum[:, 10] = entn_sum[:, 10] + get_entropy_hermite( gt0, kzind=-1, include_kz0=include_kz0) for k in range(10): kzindex = k * par['nkz0'] / 20 #print 'kzindex',kzindex entn_sum[:, k] = entn_sum[:, k] + dd.get_entropy_hermite(gt0, kzind=kzindex) entn_sum = entn_sum / float(ntime) plt.loglog(herm_grid, prefactor * entn_sum[:, 10], basex=10, basey=10) temp = prefactor * entn_sum[20, 10] temp = temp / (herm_grid**(-1))[20] plt.loglog(herm_grid, 2.0 * temp * herm_grid**(-1), '--', basex=10, basey=10, label=str(-1)) temp = prefactor * entn_sum[20, 10] temp = temp / (herm_grid**(-1.5))[20]