astro/ps2d.py: use "jet" colormap and improve axis labels

1 files changed, 4 insertions, 3 deletions

diff --git a/astro/ps2d.py b/astro/ps2d.py
index ce81b5e..ff762b6 100755
--- a/astro/ps2d.py
+++ b/astro/ps2d.py
@@ -218,11 +218,12 @@ class PS2D:
         """
         x = self.k_perp
         y = self.k_los
-        mappable = ax.pcolormesh(x[1:], y[1:], np.log10(self.ps2d[1:, 1:]))
+        mappable = ax.pcolormesh(x[1:], y[1:], np.log10(self.ps2d[1:, 1:]),
+                                 cmap="jet")
         ax.set(xscale="log", yscale="log",
                xlim=(x[1], x[-1]), ylim=(y[1], y[-1]),
-               xlabel=r"k$_{perp}$ [Mpc$^{-1}$]",
-               ylabel=r"k$_{los}$ [Mpc$^{-1}$]",
+               xlabel=r"k$_{\perp}$ [Mpc$^{-1}$]",
+               ylabel=r"k$_{||}$ [Mpc$^{-1}$]",
                title="2D Power Spectrum")
         cb = fig.colorbar(mappable, ax=ax, pad=0.01, aspect=30)
         cb.ax.set_xlabel(r"[%s$^2$ Mpc$^3$]" % self.unit)