Is there a way to make images in Python by using functions to define the colour values of individual coordinates/pixels?

I'm very new to coding and was hoping for some guidance on how to plot visualisations defined using mathematical parameters. Something along the lines of a Mandelbrot set picture or any another fractal. I'm comfortable with the mathematics involved in these I just want to know what libraries or modules I should look at to help with plotting. Then I'd like to output it to PDF. Many Thanks!