Matplotlib で等高線プロットを作成する方法

等高線プロットは、等高線を使用して 3 次元データを 2 次元で視覚化できるプロットの一種です。

次の 2 つの関数を使用して、Matplotlib で等高線プロットを作成できます。

• matplotlib.pyplot.contour() – 等高線図を作成します。
• matplotlib.pyplot.contourf() – 塗りつぶされた等高線プロットを作成します。

次の例は、これら 2 つの関数を実際に使用する方法を示しています。

例 1: Matplotlib での等高線プロット

Python に次のデータがあるとします。

 import numpy as np

x = np.linspace(0, 5, 50)
y = np.linspace(0, 5, 40)

X, Y = np.meshgrid(x, y)
Z = np.sin(X*2+Y)*3 + np.cos(Y+5)

次のコードを使用して、データの等高線プロットを作成できます。

 import matplotlib.pyplot as plt

plt. outline (X, Y, Z, colors=' black ')

プロットに 1 つの色のみが使用されている場合、破線は負の値を表し、実線は正の値を表します。

別の方法は、 cmap引数を使用してカラーマップを指定することです。レベル引数を使用して、プロットで使用するさらに多くの行を指定することもできます。

 plt. contour (X, Y, Z, levels= 30 , cmap=' Reds ') 

「Reds」cmap を使用することにしましたが、カラー パレット オプションの完全なリストはMatplotlib ドキュメント ページで見つけることができます。

例 2: Matplotlib の塗りつぶされた等高線プロット

塗りつぶされた等高線パスは、線の間のスペースが塗りつぶされることを除いて、等高線パスと似ています。

次のコードは、 contourf()関数を使用して、前の例で使用したのと同じデータの塗りつぶされた等高線プロットを作成する方法を示しています。

 plt. contourf (X, Y, Z, cmap=' Reds ') 

colorbar()関数を使用して、プロットの隣にラベル付きのカラー バーを追加することもできます。

 plt. contourf (X, Y, Z, cmap=' Reds ')
plt. colorbar () 

ここでその他の Matplotlib チュートリアルを見つけることができます。