How to plot a multivariate function in Python Matplotlib?


To plot a multivariate function in Python, we can take the following steps −

Steps

  • Set the figure size and adjust the padding between and around the subplots.

  • Create random x, y and z data points using numpy.

  • Create a figure and a set of subplots.

  • Create a scatter plot with x, y and z data points.

  • Create a colorbar for a ScalarMappable instance, s.

  • To display the figure, use show() method.

Example

import numpy as np
import matplotlib.pyplot as plt

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

def func(x, y):
   return 3 * x + 4 * y - 2 + np.random.randn(30)

x, y = np.random.randn(2, 30)
y *= 100
z = func(x, y)

fig, ax = plt.subplots()

s = ax.scatter(x, y, c=z, s=100, marker='*', cmap='plasma')

fig.colorbar(s)

plt.show()

Output

It will produce the following output −

Updated on: 01-Feb-2022

5K+ Views

Kickstart Your Career

Get certified by completing the course

Get Started
Advertisements