如何在Matplotlib中绘制平滑的二维颜色图:z=f(x,y)?
要在Matplotlib中绘制平滑的二维颜色图,其中 z = f(x, y) ,我们可以采取以下步骤−
- 设置图形大小并调整子图之间和周围的填充。
- 使用numpy创建 x 和 y 数据点。
- 使用 f(x,y) 获取 z 数据点。
- 显示数据作为图像,即在2D常规栅格上,带有z数据点。
- 要显示图像,使用 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()