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

実行結果

上記のコードを実行すると、次のような出力が得られます。