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

Làm thế nào để tạo một biểu đồ bề mặt từ một hình ảnh thang độ xám với Matplotlib?

Để tạo biểu đồ bề mặt từ hình ảnh thang độ xám bằng 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 ô con.

  • Tạo các điểm dữ liệu ngẫu nhiên bằng Numpy.

  • Nhận xx yy các điểm dữ liệu từ một raster dữ liệu hình ảnh 2d.

  • Tạo một hình mới hoặc kích hoạt một hình hiện có.

  • Lấy trục hiện tại của biểu đồ và tạo thành trục chiếu 3D.

  • Tạo biểu đồ bề mặt với cmap ='gray' .

  • Để hiển thị hình này, hãy sử dụng show () phương pháp.

Ví dụ

import numpy as np
import matplotlib.pyplot as plt

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

data = np.random.rand(5, 5)

xx, yy = np.mgrid[0:data.shape[0], 0:data.shape[1]]

fig = plt.figure()

ax = fig.gca(projection='3d')
ax.plot_surface(xx, yy, data, rstride=1, cstride=1, linewidth=0, cmap='gray')

plt.show()

Đầu ra

Làm thế nào để tạo một biểu đồ bề mặt từ một hình ảnh thang độ xám với Matplotlib?