Làm thế nào để vẽ một biểu đồ màu 2D mượt mà cho z =f (x, y) trong Matplotlib?

Để vẽ biểu đồ màu 2D mượt mà cho z =f (x, y) trong 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 x và y điểm dữ liệu bằng cách sử dụng numpy.
• Nhận z điểm dữ liệu sử dụng f (x, y) .
• Hiển thị dữ liệu dưới dạng hình ảnh, tức là trên raster 2D thông thường, với các điểm dữ liệu z.
• Để 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

def f(x, y):
   return np.array([i * i + j * j for j in y for i in x]).reshape(5, 5)

x = y = np.linspace(-1, 1, 5)
z = f(x, y)

plt.imshow(z, interpolation='bilinear')

plt.show()

Đầu ra