如何在 Matplotlib 的 streamplot 中增加色图/线宽质量？

要在 Matplotlib 的 streamplot 中增加色图/线宽质量，我们可以采取以下步骤

步骤

• 设置数字大小，并在子图之间以及子图周围调整填充。

• 创建图形和一组子图。

• 创建x 和y 数据点，然后使用 np.meshgrid() 从坐标矢量返回坐标矩阵。

• 使用x 和y 数据点找到X 和Y。

• 使用x, y, X 和Y 数据点创建一个streamplot。可以使用该方法中的linewidth 参数增加线宽。此处我们使用 linewidth=5。

• 为 ScalarMappable 实例创建颜色条，*stream.lines*。

• 使用Show() 方法显示图形。

示例

import numpy as np
from matplotlib import pyplot as plt

plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True

fig, ax = plt.subplots()

x, y = np.meshgrid(np.linspace(-5, 5, 20), np.linspace(-5, 5, 20))

X = y
Y = 3 * x - 4 * y

stream = ax.streamplot(x, y, X, Y,  density=1, linewidth=5, cmap='plasma', color=Y)

fig.colorbar(stream.lines, ax=ax)

plt.show()

输出

将产生以下输出 −

Rishikesh Kumar Rishi

更新日期：09-Oct-2021

222 次浏览

开启您的 职业

完成该课程以获得认证

开始学习