如何在 Matplotlib 中绘制 z = f(x, y) 的平滑 2D 彩色绘图？

要在 Matplotlib 中绘制 z = f(x, y) 的平滑 2D 彩色绘图，我们可以采取以下步骤 -

  • 设置图形大小，并调整子图之间和周围的填充。
  • 使用 numpy 创建 x 和 y 数据点。
  • 使用 f(x, y) 获取 z 数据点。
  • 将数据显示为图像，即，在具有 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()

输出

更新日期：2021 年 7 月 7 日

超过 3K 的浏览人数

为你的 职业做好准备

完成课程获得认证

开始
广告