How to plot a smooth 2D color plot for z = f(x, y) in Matplotlib?

To plot a smooth 2D color plot for z = f(x, y) in Matplotlib, we can take the following steps −

• Set the figure size and adjust the padding between and around the subplots.
• Create x and y data points using numpy.
• Get z data points using f(x, y).
• Display the data as an image, i.e., on a 2D regular raster, with z data points.
• To display the figure, use show() method.

Example

import numpy as np
from matplotlib import pyplot as plt

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

def f(x, y):
   return np.array([i * i + j * j for j in y for i in x]).reshape(5, 5)

x = y = np.linspace(-1, 1, 5)
z = f(x, y)

plt.imshow(z, interpolation='bilinear')

plt.show()

Output

Rishikesh Kumar Rishi

Published on 07-Jul-2021 10:47:48

• Related Questions & Answers
• Plot a 3D surface from {x,y,z}-scatter data in Python Matplotlib
• How to plot a smooth line with matplotlib?
• How to plot a 2D histogram in Matplotlib?
• Show the origin axis (x,y) in Matplotlib plot
• How do I plot multiple X or Y axes in Matplotlib?
• How to add a shared x-label and y-label to a plot created with Pandas' plot? (Matplotlib)
• How to plot 2D math vectors with Matplotlib?
• How to plot 2d FEM results using matplotlib?
• How to plot a gradient color line in matplotlib?
• How to plot a 2D matrix in Python with colorbar Matplotlib?
• Python Scatter Plot with Multiple Y values for each X
• How to a plot stem plot in Matplotlib Python?
• How to plot y=1/x as a single graph in Python?
• How to set same color for markers and lines in a Matplotlib plot loop?
• Adding caption below X-axis for a scatter plot using Matplotlib