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

Chỉ vẽ hình tam giác trên / dưới của bản đồ nhiệt trong Matplotlib

Để chỉ vẽ hình tam giác trên / dưới của bản đồ nhiệt trong matplotlib, chúng ta có thể sử dụng numpy để lấy mảng 2D có mặt nạ và chuyển chúng thành hình ảnh để tạo bản đồ nhiệt.

Các bước

  • Đặ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 dữ liệu ngẫu nhiên có kích thước 5 × 5.

  • Sử dụng numpy.tri () để tạo một mảng với 1 ở và dưới đường chéo đã cho và 0 ở những nơi khác.

  • Nhận dữ liệu mảng 2D có mặt nạ với mảng có mặt nạ (Sử dụng bước 3).

  • Sử dụng imshow () phương pháp hiển thị dữ liệu dưới dạng hình ảnh, tức là trên raster 2D thông thường.

  • Để 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.50, 3.50]
plt.rcParams["figure.autolayout"] = True
data = np.random.rand(5, 5)
mask = np.tri(data.shape[0], k=-1)
data = np.ma.array(data, mask=mask)
plt.imshow(data, interpolation="nearest", cmap='copper')
plt.show()

Đầu ra

Chỉ vẽ hình tam giác trên / dưới của bản đồ nhiệt trong Matplotlib