Matplotlib에서 z =f(x, y)에 대한 부드러운 2D 색상 플롯을 그리는 방법은 무엇입니까?

z =f(x, y)에 대한 부드러운 2D 색상 플롯을 플롯하려면 Matplotlib에서 다음 단계를 수행할 수 있습니다. -

• 그림 크기를 설정하고 서브플롯 사이 및 주변 여백을 조정합니다.
• x 만들기 및 y numpy를 사용한 데이터 포인트
• z 받기 f(x, y)를 사용하는 데이터 포인트 .
• 데이터를 z 데이터 포인트가 있는 이미지, 즉 2D 일반 래스터에 표시합니다.
• 그림을 표시하려면 show()를 사용하세요. 방법.

예

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()

출력