Problems with plotting in Nested for loop

Question

I am trying to plot in a nested loop, I expect three different plots with a different colors.

import matplotlib.pyplot as plt
import numpy as np
    
Temp = np.array([6, 7, 8])
Freq = np.arange(1, 10, 0.1)
 
# nested for loop for 3 plots
for T in Temp:
        for f in Freq:
            def quanta(f,T):
                return(f*T)
            final = quanta(f,T)  
            plt.plot(f, final)
             
plt.show()

move the plt.show()inside the for loop just after the plt.plot — Alberto
– Alberto, Commented Dec 24, 2022 at 22:03

rzz · Accepted Answer · 2022-12-24 22:21:04Z

1

one possibility is to use the subplot() function to provide three subplots. I made an adaptation in the code and from what I understand of the question, I believe it can answer you:

import matplotlib.pyplot as plt
import numpy as np

Temp = np.array([6, 7, 8])
Freq = np.arange(1, 10, 0.1)
colors = ['red', 'blue', 'green']

for i, T in enumerate(Temp):
    final = []
    for f in Freq:
        def quanta(f,T):
            return(f*T)
        final.append(quanta(f,T))
    plt.subplot(3, 1, i+1)
    plt.plot(Freq, final, color=colors[i])

plt.show()

answered Dec 24, 2022 at 22:21

rzz

776 bronze badges

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

1 Comment

Bholu Over a year ago

Thanks but I did not want subplots, so I tried the same code but using for i, T in enumerate(Temp):, I get the plot in different colour but legends now are in the same colour

Collectives™ on Stack Overflow

Problems with plotting in Nested for loop

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related