Thêm legend vào scatter plot 3D với scatter() trong Matplotlib

Để thêm legend vào scatter plot 3D với scatter() trong matplotlib, thực hiện các bước sau:

Các bước

• Thiết lập kích thước hình và padding.
• Khởi tạo biến N số mẫu.
• Tạo x, y; tạo danh sách z1, z2.
• Thêm subplot với projection='3d'.
• Vẽ x, y, z1 bằng plot() marker *.
• Vẽ x, y, z2 bằng plot() marker o.
• Đặt legend.
• Hiển thị hình bằng show().

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

Kết quả

Đồ thị 3D với hai nhóm điểm: sao (tích xy) và tròn (tổng x+y), legend góc trên phải.