Thêm chú giải trong biểu đồ phân tán 3D với scatter () trong Matplotlib

Để thêm chú giải trong biểu đồ phân tán 3D với scatter () trong matplotlib, chúng ta có thể thực hiện các bước sau -

Các bước

• Đặt kích thước hình và điều chỉnh phần đệm giữa và xung quanh các ô con.

• Khởi tạo một biến N để lưu trữ số lượng dữ liệu mẫu.

• Tạo x và y điểm dữ liệu; làm cho z1 và z2 danh sách điểm dữ liệu.

• Thêm một ô con vào hình hiện tại, với chiếu ='3d' .

• Vẽ biểu đồ x , y và z1 điểm dữ liệu bằng cách sử dụng plot () điểm trên trục 2d, với điểm đánh dấu * .

• Vẽ biểu đồ x , y và z2 điểm dữ liệu bằng cách sử dụng plot () điểm trên trục 2d, với điểm đánh dấu o .

• Đặt chú giải trên hình.

• Để hiển thị hình, chúng ta có thể sử dụng show () phương pháp.

Ví dụ

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()

Đầu ra

Nó sẽ tạo ra kết quả sau -