Mastering Matplotlib: Part 5 — Working with Images
Enhancing Visuals and Data Representation
🔙 Previous: Subplots, Layouts, and Advanced Customizations
🔜 Next: Exploring 3D Plotting
Note: You can find all the code examples for Matplotlib series in my GitHub repository.
In this section we dive into the world of image manipulation and visualization. Images are ubiquitous in data analysis and scientific research, and being able to effectively display, customize, and overlay plots on images can greatly enhance the interpretability of your results. This section will guide you through key functionalities in Matplotlib for working with images, including displaying images, customizing colormaps, adding colorbars, and overlaying plots on top of images. By the end of this tutorial, you will have the skills to handle images in Matplotlib with confidence and precision.
1. Displaying Images with imshow()
The imshow() function is the core tool in Matplotlib for displaying images. It accepts a variety of input formats, including 2D arrays for grayscale images and 3D arrays for RGB images.
Example: Displaying a Grayscale and an RGB Image
import matplotlib.pyplot as plt
import numpy as np
# Create a sample grayscale image
grayscale_image = np.random.random((10, 10))
# Create a sample RGB image
rgb_image = np.random.random((10, 10, 3))
# Display the grayscale image
plt.figure(figsize=(8, 4))
plt.subplot(1, 2, 1)
plt.imshow(grayscale_image, cmap='gray')
plt.title("Grayscale Image")
plt.axis('off')
# Display the RGB image
plt.subplot(1, 2, 2)
plt.imshow(rgb_image)
plt.title("RGB Image")
plt.axis('off')
plt.show()Output:
In this example, we generate a random grayscale image and an RGB image and display them side by side using imshow(). The cmap='gray' argument ensures the grayscale image is displayed correctly.
2. Customizing Colormaps
Colormaps are essential when visualizing data in images, as they map the data values to colors. Matplotlib offers a variety of built-in colormaps, which you can customize to enhance the visual appeal or clarity of your images.
Note: You can read more about colormaps with this link.
Example: Applying and Customizing Colormaps
# Generate sample data
data = np.random.random((10, 10))
# Display the data with different colormaps
plt.figure(figsize=(8, 6))
plt.subplot(2, 2, 1)
plt.imshow(data, cmap='viridis')
plt.title("Viridis Colormap")
plt.axis('off')
plt.subplot(2, 2, 2)
plt.imshow(data, cmap='plasma')
plt.title("Plasma Colormap")
plt.axis('off')
plt.subplot(2, 2, 3)
plt.imshow(data, cmap='coolwarm')
plt.title("Coolwarm Colormap")
plt.axis('off')
plt.subplot(2, 2, 4)
plt.imshow(data, cmap='magma')
plt.title("Magma Colormap")
plt.axis('off')
plt.show()Output:
In this example, we visualize the same data using four different colormaps: viridis, plasma, coolwarm, and magma. This helps how different colormaps can impact the perception of the data.
3. Adding Colorbars
Colorbars provide a visual reference for the mapping between data values and colors in your images. They are particularly useful when working with continuous data.
Example: Adding a Colorbar
# Generate sample data
data = np.random.random((10, 10))
# Display the image with a colorbar
plt.imshow(data, cmap='gray')
plt.colorbar(label='Intensity')
plt.title("Image with Colorbar")
plt.axis('off')
plt.show()Output:
In this example, we display an image with the gray colormap and add a colorbar to indicate the intensity values.
4. Overlaying Plots on Images
Overlaying plots on images can be useful for highlighting specific regions or data points on an image. Matplotlib allows you to easily overlay plots such as scatter plots, line plots, and more.
Example: Overlaying a Scatter Plot on an Image
# Generate a sample image and overlay data
image = np.random.random((100, 100))
x = np.random.randint(0, 100, 20)
y = np.random.randint(0, 100, 20)
plt.imshow(image, cmap='gray')
plt.scatter(x, y, color='red', marker='o')
plt.title("Scatter Plot Overlay on Image")
plt.axis('off')
plt.show()Output:
In this example, we display a grayscale image and overlay a scatter plot on top of it, with red markers indicating specific data points.
Conclusion
In this part of the “Mastering Matplotlib” series, we’ve explored various techniques for working with images in Matplotlib. From displaying images with imshow() to customizing colormaps, adding colorbars, and overlaying plots, these tools are invaluable for any data scientist or researcher working with visual data. With these skills, you can effectively convey your data's story through powerful visualizations.
In the next part of this series (Part — 6), we’ll continue our exploration by diving into more advanced topics, such as 3D plotting and interactive visualizations.
If you like the article and would like to support me make sure to:
👏 Clap for the story (as much as you liked it 😊) and follow me 👉 📰 View more content on my medium profile 🔔 Follow Me: LinkedIn | Medium | GitHub | Twitter
Feel free to share your thoughts and questions in the comments below!
