Sunday, 23 June 2024

Demonstrating different types of colormaps

 


import matplotlib.pyplot as plt

import numpy as np

# Generate sample data

data = np.random.rand(10, 10)

# List of colormaps to demonstrate

colormaps = [

    'viridis',      # Sequential

    'plasma',       # Sequential

    'inferno',      # Sequential

    'magma',        # Sequential

    'cividis',      # Sequential

    'PiYG',         # Diverging

    'PRGn',         # Diverging

    'BrBG',         # Diverging

    'PuOr',         # Diverging

    'Set1',         # Qualitative

    'Set2',         # Qualitative

    'tab20',        # Qualitative

    'hsv',          # Cyclic

    'twilight',     # Cyclic

    'twilight_shifted' # Cyclic

]

# Create subplots to display colormaps

fig, axes = plt.subplots(nrows=5, ncols=3, figsize=(15, 20))

# Flatten axes array for easy iteration

axes = axes.flatten()

# Loop through colormaps and plot data

for ax, cmap in zip(axes, colormaps):

    im = ax.imshow(data, cmap=cmap)

    ax.set_title(cmap)

    plt.colorbar(im, ax=ax)

# Adjust layout to prevent overlap

plt.tight_layout()

# Show the plot

plt.show()


Explanation:

  1. Generate Sample Data:

    data = np.random.rand(10, 10)

    This creates a 10x10 array of random numbers.

  2. List of Colormaps:

    • A list of colormap names is defined. Each name corresponds to a different colormap in Matplotlib.
  3. Create Subplots:

    fig, axes = plt.subplots(nrows=5, ncols=3, figsize=(15, 20))

    This creates a 5x3 grid of subplots to display multiple colormaps.

  4. Loop Through Colormaps:

    • The loop iterates through each colormap, applying it to the sample data and plotting it in a subplot.
  5. Add Colorbar:

    plt.colorbar(im, ax=ax)

    This adds a colorbar to each subplot to show the mapping of data values to colors.

  6. Adjust Layout and Show Plot:

    plt.tight_layout() plt.show()

    These commands adjust the layout to prevent overlap and display the plot.

Choosing Colormaps

  • Sequential: Good for data with a clear order or progression.
  • Diverging: Best for data with a critical midpoint.
  • Qualitative: Suitable for categorical data.
  • Cyclic: Ideal for data that wraps around, such as angles.

By selecting appropriate colormaps, you can enhance the visual representation of your data, making it easier to understand and interpret.


0 Comments:

Post a Comment

Popular Posts

Categories

100 Python Programs for Beginner (13) AI (33) Android (24) AngularJS (1) Assembly Language (2) aws (17) Azure (7) BI (10) book (4) Books (161) C (77) C# (12) C++ (82) Course (67) Coursera (223) Cybersecurity (24) data management (11) Data Science (127) Data Strucures (8) Deep Learning (20) Django (14) Downloads (3) edx (2) Engineering (14) Excel (13) Factorial (1) Finance (6) flask (3) flutter (1) FPL (17) Google (34) Hadoop (3) HTML&CSS (47) IBM (25) IoT (1) IS (25) Java (93) Leet Code (4) Machine Learning (53) Meta (22) MICHIGAN (5) microsoft (4) Nvidia (1) Pandas (3) PHP (20) Projects (29) Python (917) Python Coding Challenge (304) Questions (2) R (70) React (6) Scripting (1) security (3) Selenium Webdriver (2) Software (17) SQL (42) UX Research (1) web application (8)

Followers

Person climbing a staircase. Learn Data Science from Scratch: online program with 21 courses