add functionality to plotting

.gitignore

@@ -5,3 +5,5 @@
 
 # Files as produced by the tests
 tests/test_*.png
+wavelets/transform.py
+wavelets/wavelets.py

wavelets/transform.py

@@ -573,7 +573,7 @@ def coi(self):
 
         return sC, sS
 
-    def plot_power(self, ax=None, coi=True):
+    def plot_power(self, ax=None, coi=True, cmap = 'viridis'):
         """Create a basic wavelet power plot with time on the
         x-axis, scale on the y-axis, and a cone of influence
         overlaid.
@@ -586,7 +586,7 @@ def plot_power(self, ax=None, coi=True):
             fig, ax = plt.subplots()
 
         Time, Scale = np.meshgrid(self.time, self.scales)
-        ax.contourf(Time, Scale, self.wavelet_power, 100)
+        ax.contourf(Time, Scale, self.wavelet_power, 100, cmap = cmap)
 
         ax.set_yscale('log')
         ax.grid(True)

wavelets/wavelets.py

@@ -5,7 +5,7 @@
 import scipy.signal
 import scipy.optimize
 import scipy.special
-from scipy.misc import factorial
+from scipy.special import factorial
 
 __all__ = ['Morlet', 'Paul', 'DOG', 'Ricker', 'Marr', 'Mexican_hat']