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

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