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

Cách thêm subplot 3D vào figure matplotlib

Giới thiệu

Để thêm một subplot 3D vào figure matplotlib, bạn cần chỉ định projection='3d' khi thêm subplot:

  • Thiết lập kích thước hình vẽ và padding.
  • Tạo dữ liệu x, y, z bằng numpy.
  • Tạo figure mới.
  • Thêm subplot với projection='3d'.
  • Vẽ dữ liệu x, y, z bằng phương thức plot().
  • Hiển thị hình vẽ bằng show().

Ví dụ minh họa

from matplotlib import pyplot as plt
import numpy as np

# Thiết lập kích thước hình
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True

# Tạo dữ liệu x, y, z
x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
z = x ** 2 + y ** 2

fig = plt.figure()
# Thêm subplot 3D
ax = fig.add_subplot(111, projection='3d')
# Vẽ đường cong 3D
ax.plot(x, y, z, color='red', lw=7)
plt.show()

Kết quả đầu ra

Đường cong 3D parabol hình chữ U:

Cách thêm subplot 3D vào figure matplotlib