CODEWITHSUNDEEP
python-3.xmatplotlib
Laveena
Laveena

Reputation: 471

Plot dynamic heatmap/ color map in while loop Python

I am working on a code to plot a heat map using matplotlib library,the problem I am facing is , instead of my (same) plot being updated for every new set of data in the loop I am getting new window of plot for each data set. I referred this good post about it but it didn't help me, may be because of limited knowledge of drawing graphs in python.Any help and suggestions in this regard will be really helpful!

Please find my code below (displaying multiple plot window) below:

import numpy as np
import scipy.io.wavfile
import wave
import matplotlib.pyplot as plt

plt.rcParams["figure.figsize"] = 5,2

rate, data = scipy.io.wavfile.read('test1.wav')
chunk = np.array_split(data, 682)
l = len(chunk)
count= 0
while(count<l):
    x = np.linspace(1,len(chunk[count]),num =len(chunk[count]) )
    y= np.array(chunk[count])

    fig, (ax, ax2) = plt.subplots(nrows=2, sharex=True)
    extent = [x[0] - (x[1] - x[0]) / 2., x[-1] + (x[1] - x[0]) / 2., 0, 1]
    ax.imshow(y[np.newaxis, :], cmap="plasma", aspect="auto", extent=extent)
    ax.set_yticks([])
    ax.set_xlim(extent[0], extent[1])
    ax2.plot(x,y)
    plt.tight_layout()
    plt.draw()
    plt.pause(0.1)
    count = count +1

Upvotes: 2

Views: 3592

Answers (1)

Laveena
Laveena

Reputation: 471

Well, It will be my 2nd answer to my own question :P I figured it out!

Just move this line: fig, (ax, ax2) = plt.subplots(nrows=2, sharex=True)above while loop,so that you initialize one figure and ask this figure to be constant while the values in the plot keeps on changing.

Do addplt.cla() in the end , within the while loop so you clear up the plot from the old data else the new plot will be overwritten/plotted over old one's.

Upvotes: 1

Related Questions