Displaying a FITS image. Show IDL code show Python code. Making annotations on a FITS image. See photos, profile pictures and albums from PLOTS.
A short tutorial on plotting images with Matplotlib.
First, let's start IPython. It is a most excellent enhancement to thestandard Python prompt, and it ties in especially well withMatplotlib. Start IPython either directly at a shell, or with the JupyterNotebook (where IPython as a running kernel).
With IPython started, we now need to connect to a GUI event loop. Thistells IPython where (and how) to display plots. To connect to a GUIloop, execute the %matplotlib magic at your IPython prompt. There's moredetail on exactly what this does at IPython's documentation on GUIevent loops.
If you're using Jupyter Notebook, the same commands are available, butpeople commonly use a specific argument to the %matplotlib magic:
This turns on inline plotting, where plot graphics will appear in yournotebook. This has important implications for interactivity. For inline plotting, commands incells below the cell that outputs a plot will not affect the plot. For example,changing the color map is not possible from cells below the cell that creates a plot.However, for other backends, such as Qt5, that open a separate window,cells below those that create the plot will change the plot - it is alive object in memory.
This tutorial will use matplotlib's imperative-style plottinginterface, pyplot. This interface maintains global state, and is veryuseful for quickly and easily experimenting with various plotsettings. The alternative is the object-oriented interface, which is alsovery powerful, and generally more suitable for large applicationdevelopment. If you'd like to learn about the object-orientedinterface, a great place to start is our Usage guide. For now, let's get onwith the imperative-style approach:
Loading image data is supported by the Pillow library. Natively, Matplotlibonly supports PNG images. The commands shown below fall back on Pillow ifthe native read fails.
The image used in this example is a PNG file, but keep that Pillowrequirement in mind for your own data.
Here's the image we're going to play with:
It's a 24-bit RGB PNG image (8 bits for each of R, G, B). Dependingon where you get your data, the other kinds of image that you'll mostlikely encounter are RGBA images, which allow for transparency, orsingle-channel grayscale (luminosity) images. You can right click onit and choose 'Save image as' to download it to your computer for therest of this tutorial.
And here we go...
Out:
Note the dtype there - float32. Matplotlib has rescaled the 8 bitdata from each channel to floating point data between 0.0 and 1.0. Asa side note, the only datatype that Pillow can work with is uint8.Matplotlib plotting can handle float32 and uint8, but imagereading/writing for any format other than PNG is limited to uint8data. Why 8 bits? Most displays can only render 8 bits per channelworth of color gradation. Why can they only render 8 bits/channel?Because that's about all the human eye can see. More here (from aphotography standpoint): Luminous Landscape bit depth tutorial.
Each inner list represents a pixel. Here, with an RGB image, thereare 3 values. Since it's a black and white image, R, G, and B are allsimilar. An RGBA (where A is alpha, or transparency), has 4 valuesper inner list, and a simple luminance image just has one value (andis thus only a 2-D array, not a 3-D array). For RGB and RGBA images,matplotlib supports float32 and uint8 data types. For grayscale,matplotlib supports only float32. If your array data does not meetone of these descriptions, you need to rescale it.
So, you have your data in a numpy array (either by importing it, or bygenerating it). Let's render it. In Matplotlib, this is performedusing the imshow()
function. Here we'll grabthe plot object. This object gives you an easy way to manipulate theplot from the prompt.
You can also plot any numpy array.
Pseudocolor can be a useful tool for enhancing contrast andvisualizing your data more easily. This is especially useful whenmaking presentations of your data using projectors - their contrast istypically quite poor.
Pseudocolor is only relevant to single-channel, grayscale, luminosityimages. We currently have an RGB image. Since R, G, and B are allsimilar (see for yourself above or in your data), we can just pick onechannel of our data:
Out:
Now, with a luminosity (2D, no color) image, the default colormap (aka lookup table,LUT), is applied. The default is called viridis. There are plenty ofothers to choose from.
Out:
Note that you can also change colormaps on existing plot objects using theset_cmap()
method:
Note
However, remember that in the Jupyter Notebook with the inline backend,you can't make changes to plots that have already been rendered. If youcreate imgplot here in one cell, you cannot call set_cmap() on it in a latercell and expect the earlier plot to change. Make sure that you enter thesecommands together in one cell. plt commands will not change plots from earliercells.
There are many other colormap schemes available. See the list andimages of the colormaps.
It's helpful to have an idea of what value a color represents. We cando that by adding a color bar to your figure:
Out:
Sometimes you want to enhance the contrast in your image, or expandthe contrast in a particular region while sacrificing the detail incolors that don't vary much, or don't matter. A good tool to findinteresting regions is the histogram. To create a histogram of ourimage data, we use the hist()
function.
Out:
Most often, the 'interesting' part of the image is around the peak,and you can get extra contrast by clipping the regions above and/orbelow the peak. In our histogram, it looks like there's not muchuseful information in the high end (not many white things in theimage). Let's adjust the upper limit, so that we effectively 'zoom inon' part of the histogram. We do this by passing the clim argument toimshow. You could also do this by calling theset_clim()
method of the image plotobject, but make sure that you do so in the same cell as your plotcommand when working with the Jupyter Notebook - it will not changeplots from earlier cells.
You can specify the clim in the call to plot
.
You can also specify the clim using the returned object
Out:
Interpolation calculates what the color or value of a pixel 'should'be, according to different mathematical schemes. One common placethat this happens is when you resize an image. The number of pixelschange, but you want the same information. Since pixels are discrete,there's missing space. Interpolation is how you fill that space.This is why your images sometimes come out looking pixelated when youblow them up. The effect is more pronounced when the differencebetween the original image and the expanded image is greater. Let'stake our image and shrink it. We're effectively discarding pixels,only keeping a select few. Now when we plot it, that data gets blownup to the size on your screen. The old pixels aren't there anymore,and the computer has to draw in pixels to fill that space.
We'll use the Pillow library that we used to load the image also to resizethe image.
Here we have the default interpolation, bilinear, since we did notgive imshow()
any interpolation argument.
Let's try some others. Here's 'nearest', which does no interpolation.
and bicubic:
Bicubic interpolation is often used when blowing up photos - peopletend to prefer blurry over pixelated.
Total running time of the script: ( 0 minutes 1.183 seconds)
Keywords: matplotlib code example, codex, python plot, pyplotGallery generated by Sphinx-Gallery