Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Vẽ mặt phẳng 3D từ dữ liệu phân tán {x,y,z} trong Python Matplotlib

Để vẽ mặt phẳng 3D từ dữ liệu phân tán x, y, z trong Python, ta thực hiện:

  • Thiết lập kích thước hình và điều chỉnh khoảng đệm giữa và xung quanh subplots.
  • Tạo hình mới hoặc kích hoạt hình hiện có dùng figure().
  • Thêm trục vào hình như phần của subplot.
  • Tạo x, y, X, Y, Z dùng numpy.
  • Vẽ x, y, z dùng plot_surface().
  • Hiển thị hình dù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

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
x = np.array(np.linspace(-2, 2, 100))
y = np.array(np.linspace(-2, 2, 10))
X, Y = np.meshgrid(x, y)
Z = X * np.exp(-X ** 2 - Y ** 2)
ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap="plasma",
                linewidth=0, antialiased=False)
plt.show()

Kết quả

Vẽ mặt phẳng 3D từ dữ liệu phân tán {x,y,z} trong Python Matplotlib