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

PythonServer Side ProgrammingProgramming

#### Beyond Basic Programming - Intermediate Python

Most Popular

36 Lectures 3 hours

#### Practical Machine Learning using Python

Best Seller

91 Lectures 23.5 hours

#### Practical Data Science using Python

22 Lectures 6 hours

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

Updated on 07-Jul-2021 10:48:33