import os
import sys
sys.path.insert(0, "/home/phsun/PyPSI")
import PyPSI as psi
import matplotlib.pyplot as plt
import numpy as np
import tables
fig, ax = plt.subplots()
f = tables.open_file("density0000.h5", mode = 'r')
density = np.array(f.get_node("/density"))
f.close()
## calculate power spectrum
power, k = psi.powerSpectrum(density, \
dims=(300., 300., 300.), bins = 256)
l = 2*np.pi/k
## plot spectrum
ax.loglog(l, power, label = "0000")
## save to file
np.savez_compressed("spectrum0000.npz", power=power, l=l)
ax.set_xlabel("l (Mpc/h)")
ax.set_ylabel("power")
legend = plt.legend(loc = 'upper right', shadow = True)
fig.savefig("spectrum_PSI_0000.png")
sys.path.insert(0, "/home/phsun/PyPSI")
import PyPSI as psi
import matplotlib.pyplot as plt
import numpy as np
import tables
fig, ax = plt.subplots()
i = 0
while os.path.isfile("density%04d_64.h5" % i):
f = tables.open_file("density%04d_64.h5" % i, mode="r")
density = np.array(f.get_node("/density"))
f.close()
## calculate power spectrum
power, k = psi.powerSpectrum(density, dims=(300.0 / 4, 300.0 / 4, 300.0 / 4), bins=100)
l = 2 * np.pi / k
## plot spectrum
ax.loglog(l, power, label="%04d" % i)
## save to file
np.savez_compressed("spectrum%04d_64.npz" % i, power=power, l=l)
i += 1
ax.set_xlabel("l (Mpc/h)")
ax.set_ylabel("power")
legend = plt.legend(loc="upper right", shadow=True)
fig.savefig("spectrum_PSI_64.png")
