PythonのMatplotlibでワイヤーフレームプロットを作成する方法をわかりやすく解説

Matplotlibは、Pythonで最も広く使われているデータ可視化ライブラリのひとつです。データの可視化は、大量の数値を一つひとつ確認したり複雑な計算を行ったりしなくても、データ全体の傾向や構造を直感的に把握できるようにする重要なステップです。また、定量的な洞察を聴衆に効果的に伝えるための手段としても非常に有用です。

Matplotlibは主に2次元のプロットを作成するために使用されますが、オブジェクト指向のAPIを備えており、作成したグラフをPythonアプリケーション内に簡単に組み込むことができます。IPythonシェル、Jupyter Notebook、SpyderなどのIDEとも連携可能です。Matplotlib自体はPythonで書かれており、数値計算ライブラリであるNumPyを基盤として構築されています。

Matplotlibのインストール方法

Windows環境では、以下のコマンドを実行するだけでMatplotlibをインストールできます。

pip install matplotlib

Matplotlibの依存パッケージ

Matplotlibを動作させるには、以下のパッケージが必要です。

Python(バージョン3.4以上)
NumPy
Setuptools
Pyparsing
Libpng
Pytz
FreeType
Six
Cycler
Dateutil

ワイヤーフレームプロットの作成手順

それでは、Matplotlibを使ってワイヤーフレームプロットを作成する具体的な方法を見ていきましょう。

サンプルコード

from mpl_toolkits import mplot3d
import numpy as np
import matplotlib.pyplot as plt

def my_fun(x, y):
    return np.sin(np.sqrt(x ** 4 + y ** 4))

x = np.linspace(-8, 8, 30)
y = np.linspace(-8, 8, 30)

X, Y = np.meshgrid(x, y)
Z = my_fun(X, Y)

fig = plt.figure()
ax = plt.axes(projection='3d')
ax.plot_wireframe(X, Y, Z, color='red')
plt.xlabel('X axis')
plt.ylabel('Y axis')
ax.set_title('A wireframe plot')
plt.show()

実行結果

このコードを実行すると、赤色のワイヤーフレームで描かれた3次元曲面が表示されます。波打つような形状の曲面が、格子状の線で表現されているのが確認できます。

コードの解説

• 必要なパッケージ(mpl_toolkits、NumPy、matplotlib.pyplot)をインポートし、それぞれ別名を付けて扱いやすくしています。

• 「my_fun」という関数を定義し、サイン関数を使ってプロット用のデータを生成します。この関数は、xとyの値からz座標を計算します。

• NumPyの「linspace」メソッドを使用して、-8から8までの範囲を30等分した等間隔の数列を生成します。

• 「meshgrid」でX座標とY座標のグリッドを作成し、先ほど定義した関数を呼び出してZ座標の値を算出します。

• Figureオブジェクトを作成し、Axesオブジェクトの投影法(projection)を「3d」に指定することで、3次元プロットの準備を行います。

• Matplotlibが提供する「plot_wireframe」関数を呼び出し、線の色を赤に指定してワイヤーフレームを描画します。

• X軸・Y軸のラベルとタイトルを設定した後、「show」関数を使ってプロットを画面に表示します。

このように、Matplotlibとmpl_toolkitsのmplot3dを組み合わせることで、わずか十数行のコードで美しい3次元ワイヤーフレームプロットを作成できます。色や線のスタイルを変更すれば、さまざまな表現にも対応可能です。