Source

stockwelltransform / stockwell / plot.py

Full commit
import matplotlib.pyplot as plt

from matplotlib.offsetbox import TextArea, DrawingArea, OffsetImage, \
     AnnotationBbox

from mpl_toolkits.axes_grid1 import host_subplot


def stpowerstack(x,stx):
    """
    need to add row labels
    """
    ax1=host_subplot(211)
    plt.plot(x)
    ax2 = host_subplot(212)
    #pax2 = ax2.twinx()
    #pax2.set_ylabel('frequency(Hz)')
    plt.imshow(abs(stx), aspect='auto')
    #plt.ylabel('frequency(Hz)')
    plt.ylabel('strow(f*L/fs)')
    return ax2