Python/SciPyで積分(1重/2重)した結果を可視化する方法

  • このエントリーをはてなブックマークに追加

Pythonのscipy.integrate.quadやscipy.integrate.dblquadを使えば簡単に数値積分ができますが、当然結果はシンプルに積分値が返ってくるだけです。ここでは他者へ説明する目的で基本的な1重積分をはじめ、2重積分までのmatplotlibによる可視化方法を紹介します。

こんにちは。wat(@watlablog)です。今回は積分の可視化をテーマにscipy.integrateを使ってみます!

本記事の目標

積分結果を可視化する

曲線を積分すると面積を、曲面を2重積分すると体積を求めることができます。
一般的に積分がどのような処理を行っているかを他者へ説明する時、下図に示すような、関数と求積した部分を共にプロットするのではないでしょうか。

積分の可視化結果例
Advertisements

今回はPythonのscipy.integrateによる積分方法を確認すると共に、matplotlibによる可視化例を得るところまでを扱います。まず初めに上図と同様の最も簡単な1変数関数の1重積分を扱い、以下の結果を出すコードを書きます。

Pythonによる積分の可視化例1

次に曲面を2重積分して体積を求める問題に対し、色々な関数で可視化を試してみます。可視化の方法は色々あれど、下図のようなアウトプットを出すコードを紹介します。これはガウス関数の2重積分区間に色付けしたり、区間がわかりやすくなるように黒いエッジをつけているものです。

Pythonによる積分の可視化例2

それでは早速コードを書いてみましょう!

関連記事

Pythonによる積分は以下の記事も参考にしてください。

Pythonでルジャンドル多項式を使ってガウス積分をする方法
積分で解くWi-FiパスワードをPython/sympyで計算する

動作環境

この記事のコードは以下のPython環境で動作を確認しました。

Python Python 3.9.6
PyCharm (IDE) PyCharm CE 2020.1
Numpy 1.23.0
Scipy 1.4.1
matplotlib 3.7.1

1重積分の可視化を行うPythonコード

まずは最も簡単な式(1)を使って積分と可視化を行ってみましょう。

\[ f(x)=x \tag{1} \]

scipyのintegrateからquadをimportすることで積分を行います。関数fを定義しておき、積分区間a, bを使ってquad(f, a, b)と書くことで求積が可能です。

積分結果の可視化は.fill_betweenで行っています。1重積分の可視化はこれ一択だと思いました。

scipy.integrate.quadの公式ドキュメント:https://docs.scipy.org/doc/scipy/reference/generated/scipy.integrate.quad.html

結果はこちら。出力ウィンドウにもquadの結果が書かれていますが、labelにも載せるようにしてみました。手計算してみるとわかりますが、quadの数値1.5は正解です(この積分は暗算でもできますね)。

積分結果付き可視化結果

次はもう少し変曲点が多くて面白い関数として、式(2)で試してみましょう。

\[ f(x)= x^{4} - 10 x^{3} + 25 x^{2} \tag{2} \]

関数fの部分と、積分区間a, bを変更します。

こちらが結果です。式を変更するだけで正常に機能していることがわかりました。

積分結果付き可視化結果2

これで1重積分の計算方法や可視化方法はわかったので、次に進みましょう。

2重積分の可視化を行うPythonコード

次は式(3)の2重積分を行います。式(3)は平面の方程式です。ここで\(w_{0}=1, w_{1}=1, w_{2}=1\)とします。

\[ f(x, y) = w_{0} + w_{1} x + w_{2} y \tag{3} \]

コードはこちら。2重積分はscipy.integrate.dblquadを使います(double quadratureという意味らしい)。2重積分の可視化は色々試行錯誤しましたが、積分区間を黒いエッジで囲い、面にカラーマップを貼るという見せ方が最も美しいとの結論に達しました(個人の感想です)。

scipy.integrate.dblquadの公式ドキュメント:https://docs.scipy.org/doc/scipy/reference/generated/scipy.integrate.dblquad.html

積分区間をエッジで囲うために.plot_wireframeを使っていますが、これだけだと細かい線が多数表示されるためrstrideとcstrideを使ってメッシュの間隔を大きくする工夫をしました。

また本筋とは関係ないですが、*x_range_integrateとアスタリスクをつけて引数を渡すことで一つずつ記載しなくて良いという小ワザも使ってみました。

結果がこちら。3次元プロットはやはり映えますね。積分結果は175(数値積分なのでピッタリいかない)となりました。果たしてこの結果は合っているのでしょうか?2重積分の暗算は少々やっかいなので計算してみましょう。

積分結果付き可視化結果3

積分は加算に関して線形であるため、3つの積分に分解できます。3つ目の項は奇関数であるため0になり簡単になるため、式(4)のように展開して答えは175を得ました。プログラムは正しく動作しているようですね。

\[ \begin{eqnarray} \int_{0}^{5}\int_{-5}^{5} (1 + x + y) dydx &=& \int_{0}^{5} \int_{-5}^{5} 1 \, dy \, dx + \int_{0}^{5} \int_{-5}^{5} x \, dy \, dx + \int_{0}^{5} \int_{-5}^{5} y \, dy \, dx\\ &=&5 \times 10 + \frac{5^2}{2} \times 10 + 0\\ &=&175 \end{eqnarray} \tag{4} \]

自分でプログラムを書いた時は、手計算でも簡単に答えが出る問題を解いて検証することが重要です!今回のケースだとxとyの範囲を逆にしてしまうバグもあるあるです。

次は式(5)で示す2次元のガウス関数です。2次元ガウス関数は様々な用途で用いられるので、これも2重積分してみましょう。

\[ f(x, y) = e^{-\frac{(x - \mu_x)^2}{2 \sigma_x^2}} e^{-\frac{(y - \mu_y)^2}{2 \sigma_y^2}} \tag{5} \]

Pythonコードで変更するのは関数部分のみです。積分範囲は先ほどと同様にしました。

結果がこちら。綺麗ですね。ガウス関数は天文学であれば星の光強度、工学であればベクトルデータのヒストグラムといった用途で2重積分の需要がありそうです。

積分結果付き可視化結果4

まとめ

この記事ではscipy.integrate.quadやdblquadを使った積分方法を紹介しました。それだけだと類似の記事がたくさんあったので、このブログでは主に可視化に特化してみました。

積分結果を得るだけなら可視化は全く必要ありませんが、結果を他者へ説明したり理解を深めたりする時には有用だと思います。

今回は積分区間が定数の問題のみを扱いましたが、世の中には積分区間にxやyが含まれる問題もあります。次回はそのような一歩進んだ問題の可視化もトライしてみたいと思いました。

基本的な積分を求めるPythonコードが書けました!
X(旧Twitter)でも関連情報をつぶやいているので、wat(@watlablog)のフォローお待ちしています!

  • このエントリーをはてなブックマークに追加

SNSでもご購読できます。

コメントを残す

*