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

Tạo cốt truyện 3D trong Matplotlib từ mảng 3D numpy

Để tạo một âm mưu 3D từ một mảng 3D numpy, chúng ta có thể tạo một mảng 3D bằng cách sử dụng numpy và trích xuất các điểm x, y và z.

  • 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 bằng cách sử dụng add_subplot () phương pháp.
  • Tạo dữ liệu ngẫu nhiên có size =(3, 3, 3) .
  • Trích xuất dữ liệu x, y và z từ mảng 3D.
  • Vẽ đồ thị các điểm phân tán 3D trên trục đã tạo
  • Để 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.00, 3.50]
plt.rcParams["figure.autolayout"] = True
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
data = np.random.random(size=(3, 3, 3))
z, x, y = data.nonzero()
ax.scatter(x, y, z, c=z, alpha=1)
plt.show()

Đầu ra

Tạo cốt truyện 3D trong Matplotlib từ mảng 3D numpy