How to continue plotting in the same window in python?

Question

I am having a problem figuring out how to keep plotting in the same plot window (I want my plots after a single plot to be done in the same window, i.e. I don't want to close the window to get to other plot).Could I use the arrows at the bottom of the plot window to switch to next plot? Here is my code :

for iteration in range(0, max_iters):
    idx = findClosestCentroids(X, centroids)
    centroids = computeCentroids(X, idx, K)

    if plot is True:
        data = c_[X, idx]
        for i in range(1, K + 1):
            data_1 = data[data[:, n] == i]
            pyplot.plot(data_1[:, 0], data_1[:, 1], linestyle=' ', color=dict[i - 1], marker='o', markersize=3)

        pyplot.plot(centroids[:, 0], centroids[:, 1], 'k*', markersize=15)
        pyplot.show(block=True)
        pyplot.hold(True)

Here, data is an mXn+1 matrix and the nth column has values ranging from 1 to K, centroids is a kXn matrix and idx is an mX1 matrix

Paul H · Accepted Answer · 2015-10-16 17:49:04Z

1

Never use pyplot to draw anything. The only thing it's really good for is creating figures, axes, and some artists.

Without running your example, I do:

for iteration in range(0, max_iters):
    fig, ax = plt.subplots()
    idx = findClosestCentroids(X, centroids)
    centroids = computeCentroids(X, idx, K)

    if plot is True:
        data = c_[X, idx]
        for i in range(1, K + 1):
            data_1 = data[data[:, n] == i]
            ax.plot(data_1[:, 0], data_1[:, 1], linestyle=' ', color=dict[i - 1], marker='o', markersize=3)

        ax.plot(centroids[:, 0], centroids[:, 1], 'k*', markersize=15)

    fig.show()

edited Oct 16, 2015 at 17:49

answered Oct 16, 2015 at 17:37

Paul H

68.7k23 gold badges165 silver badges139 bronze badges

Sign up to request clarification or add additional context in comments.

8 Comments

Rochak Gupta Over a year ago

Thanks for the tip! But is it possible to plot the resultant plot for each outer loop iteration in the same window? The given code just gives a single (final) plot.

Paul H Over a year ago

sure, just move the plot creating into the first loop (I'll edit)

Rochak Gupta Over a year ago

It did the trick! Thanks! However, the windows all opened without a delay and even closed before I could look at them.

Paul H Over a year ago

Hmm, I never look at interactive figures (I just save them to PNGs). What happens if you feed num=iteration to plt.subplots?

Rochak Gupta Over a year ago

It worked the same way it did before I added num=iteration to plt.subplots.

|

Collectives™ on Stack Overflow

How to continue plotting in the same window in python?

1 Answer 1

8 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

8 Comments

Your Answer

Sign up or log in

Post as a guest

Related