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

Vẽ một biểu đồ bề mặt bị che bằng Python, Numpy và Matplotlib

Để vẽ biểu đồ bề mặt bị che bằng Python, Numpy và Matplotlib, 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ó.
  • Thêm 'cái rìu' vào hình như một phần của sắp xếp ô phụ.
  • Trả về ma trận tọa độ từ vectơ tọa độ, pi và theta.
  • Tạo x, y và z bằng các điểm dữ liệu bị che.
  • Tạo biểu đồ bề mặt với các điểm dữ liệu x, y và z.
  • Để 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.00, 3.50]
plt.rcParams["figure.autolayout"] = True

fig = plt.figure()
ax = fig.add_subplot(111, projection="3d")
pi, theta = np.meshgrid(
   np.arange(1, 10, 2) * np.pi / 4,
   np.arange(1, 10, 2) * np.pi / 4)

x = np.cos(pi) * np.sin(theta)
y = np.sin(pi) * np.sin(theta)
z = np.ma.masked_where(x >= 0.01, y)

ax.plot_surface(x, y, z, color='red')

plt.show()

Đầu ra

Nó sẽ tạo ra kết quả sau

Vẽ một biểu đồ bề mặt bị che bằng Python, Numpy và Matplotlib Vẽ một biểu đồ bề mặt bị che bằng Python, Numpy và Matplotlib