Computer >> Máy Tính >  >> Lập trình >> Python

Lô đất Matplotlib 3D plot_surface với phép chiếu đường viền

Để vẽ biểu đồ 3D plot_surface với phép chiếu đường viền, chúng ta có thể sử dụng plot_surface () contourf () phương pháp.

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.

  • Tạo x, y, X, Y Z điểm dữ liệu bằng cách sử dụng numpy.

  • Tạo một hình mới hoặc kích hoạt một hình hiện có bằng cách sử dụng figure () phương pháp.

  • Thêm '~ .axes.Axes' vào hình như một phần của sắp xếp ô con, với phép chiếu 3D.

  • Sử dụng plot_surface () phương pháp tạo biểu đồ bề mặt.

  • Tạo một đồ thị đường viền được tô 3D bằng cách sử dụng contourf () phương pháp.

  • Rời khỏi trục.

  • Để hiển thị hình này, hãy sử dụng show () phương pháp.

Ví dụ

import numpy as np
from matplotlib import pyplot as plt

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

x = np.arange(-5, 5, 0.1)
y = np.arange(-5, 5, 0.1)
X, Y = np.meshgrid(x, y)
Z = X * np.exp(-X - Y)

fig = plt.figure()

ax = fig.add_subplot(111, projection='3d')
surf = ax.plot_surface(X, Y, Z, rstride=8, cstride=8, alpha=1, cmap="copper")
ax.contourf(X, Y, Z, zdir='z', offset=np.min(Z), cmap="plasma")
ax.contourf(X, Y, Z, zdir='x', offset=-5, cmap="PiYG_r")
ax.contourf(X, Y, Z, zdir='y', offset=5, cmap="PuBuGn")

ax.axis('off')
plt.show()

Đầu ra

Lô đất Matplotlib 3D plot_surface với phép chiếu đường viền