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

Vẽ bề mặt 3D từ dữ liệu vật chất {x, y, z} trong Python Matplotlib

Để vẽ bề mặt 3D từ x , y z phân tán dữ liệu trong Python, chúng ta có thể thực hiện các bước sau -

  • Đặt kích thước hình và điều chỉnh phần đệm giữa và xung quanh các ô phụ.
  • 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 các trục vào hình như một phần của sắp xếp ô phụ.
  • Tạo x , y , X , Y Z điểm dữ liệu bằng cách sử dụng numpy.
  • Lô đất x , y z điểm dữ liệu sử dụng plot_surface () phương pháp.
  • Để hiển thị hình này, hãy sử dụng show () phương pháp.

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

Đầu ra

Vẽ bề mặt 3D từ dữ liệu vật chất {x, y, z} trong Python Matplotlib