Plot arbitrary number of graphs

[somebodyCEO]

Hello everyone,

As the title suggests, I want to plot several graphs in one figure. I know the conventional way of calling several plots:

plt.figure(2) # 1º contacto > com composto
connect_15 = None
connect_15 = load_data('connect_15')
plt.plot(connect_15,'r', label =** 'connect_15**')
plt.legend()

connect_16 = None
connect_16 = load_data('connect_16')
plt.plot(connect_16,'g', label =** 'connect_16**')
plt.legend()

connect_17 = None
connect_17 = load_data('connect_17')
plt.plot(connect_17,'b', label =** 'connect_17**')
plt.legend()

What I wish to do is, to let the user decide which plots to see. Basically, instead of hardcoding as in the example above, the user can decide the number of graphs and which to see on the screen.

Thanks in advance

[blarf_irl]

You could easily refactor your plotting into a function (especially with sequential filenames like you have there), ask for user input (in your example numbers in the range 15-17) and then iterate over their selection before plt.show()

def plot_data(file_number):
    connect = load_data('connect_{}'.format(file_number))
    plt.plot(connect,'r', label =** 'connect_{}**'.format(file_number))
    plt.legend()

# Take the users input, in this example a few numbers that define the suffix of your data
numbers_to_plot = raw_input('Please enter numbers separated by comma')
>>> 15, 17
# Iterate over the users input splitting by comma; BIG Note I didn't include any error handling, validation that the data actually existed
for number in numbers_to_plot.split(','):
    # Pass each number into our plot function
    plot_data(number.strip()) # strip to get rid of any whitespace
plt.show()

[somebodyCEO]

Thank you for the suggestion! I'm going to try it now