Add a legend in a 3D scatterplot with scatter() in Matplotlib

MatplotlibPythonData Visualization

To add a legend in a 3D scatterplot with scatter() in matplotlib, we can take the following steps −

Steps

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

  • Initialize a variable N to store the number of sample data.

  • Create x and y data points; make z1 and z2 data points list.

  • Add a subplot to the current figure, with projection='3d'.

  • Plot the x, y and z1 data points using plot() points on 2d axes, with marker *.

  • Plot the x, y and z2 data points using plot() points on 2d axes, with marker o.

  • Place legend on the figure.

  • To display the figure we can use show() method.

Example

import matplotlib.pyplot as plt
import numpy as np

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

N = 100

x = np.random.rand(N)
y = np.random.rand(N)

z1 = [i*j for (i, j) in zip(x, y)]
z2 = [i+j for (i, j) in zip(x, y)]

axes = plt.subplot(111, projection='3d')

axes.plot(x, y, z1, "*", label="xy")
axes.plot(x, y, z2, "o", label="x+y")

plt.legend(loc="upper right")

plt.show()

Output

It will produce the following output −

raja
Updated on 01-Feb-2022 11:52:09

Advertisements