Màu đường cong tham số 3D trong Matplotlib.pyplot

Để vẽ đường cong tham số (parametric curve) 3D với màu thay đổi theo tọa độ, ta dùng ax.scatter() với colormap trên axes 3D.

Các bước

• Thiết lập figure size và autolayout.
• Tạo figure và axes 3D.
• Tính tham số: theta, z, r, x, y cho đường xoắn 3D.
• Vẽ scatter 3D với c=x, cmap="copper" để màu thay đổi theo x.
• Đặt tiêu đề.
• Hiển thị bằng show().

Ví dụ

import numpy as np
import matplotlib.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')
theta = np.linspace(-4 * np.pi, 4 * np.pi, 100)
z = np.linspace(-2, 2, 100)
r = z ** 2 + 1
x = r * np.sin(theta)
y = r * np.cos(theta)
ax.scatter(x, y, z, c=x, cmap="copper")
ax.set_title("Parametric Curve")
plt.show()

Kết quả