How can I save a matplotlib plot that is the output of a function in jupyter?

Question

I\'m actually working on a project using google colaboratory. I\'m using the pygad module(a Genetic Algorithm module). After running the algorithm, one can obtain the plot of a