3D Scatter Plot พร้อม Hue Colormap และ Legend ใน Matplotlib

ใช้ scatter() บน axes 3D พร้อม cmap และ legend_elements() สำหรับ legend

ตัวอย่างโค้ด

import numpy as np
import seaborn as sns
from matplotlib import pyplot as plt
from matplotlib.colors import ListedColormap

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

x = np.random.rand(100)
y = np.random.rand(100)
z = np.random.rand(100)

fig = plt.figure()
ax = fig.gca(projection='3d')
cmap = ListedColormap(sns.color_palette("husl", 256).as_hex())
sc = ax.scatter(x, y, z, s=40, c=x, marker='o', cmap=cmap, alpha=1)
plt.legend(*sc.legend_elements(), bbox_to_anchor=(1.05, 1), loc=2)
plt.show()

ผลลัพธ์

3D scatter plot สีตามค่า x พร้อม legend ด้านขวา