How to plot scatter points in a 3D figure with a colorbar in Matplotlib?

MatplotlibPythonData Visualization

To plot scatter points in a 3D figure with a colorbar in matplotlib, we can use the scatter() and colorbar() methods.

Steps

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

  • Create a new figure or activate an existing figure using figure() method.

  • Add an axis as a subplot arrangement.

  • Create xs, ys and zs data points using numpy.

  • Use scatter() method to create a scatter plot.

  • Use colorbar() method with scatter scalar mappable instance for colorbar.

  • 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
fig = plt.figure()
ax = fig.add_subplot(projection="3d")
xs = np.random.rand(100)
ys = np.random.rand(100)
zs = np.random.rand(100)
sc = ax.scatter(xs, ys, zs, c=xs, cmap="copper")
plt.colorbar(sc)
plt.show()

Output

raja
Published on 03-Jun-2021 12:54:01
Advertisements