はじめに

こんにちは、leisurelyです。この記事では、経済学を勉強していくうえで「コブダグラス型生産関数」に次いで逃れることのできない「ces型生産関数」を3Dグラフを用いて可視化します。
以前に投稿した「コブダグラス型生産関数の可視化」が思いのほか見ていただけてるようなのでces型もまとめました。

この記事を読んで頂くにあたり注意点をいくつか述べさせていただきます。
・~~私はただの経済学部生です（期待しないでね）~~下書きを書いていた時は学部生だったのに今では大学院生に…
・間違っている可能性は十分あります
・ほぼ自分のためのメモ
以上のことをご了承いただいた上で読んで頂けるとありがたいです。
間違っていることや誤解を生みそうな表現がある場合にはコメントの方よろしくお願いします。また、それら以外のコメントも受け付けていますのでお気軽にコメントください。

※この記事はプログラミングを学ぶのではなく、プログラミングというツールを用いて経済学をより理解するのが目的です。また、プログラミングわからないけれど結果だけ見たいという方のために、最後に結果を載せています。

ces型生産関数について

念のため、ces型生産関数について簡単にざっくりと解説しておきます。Constant elasticity of substitution(ces)型生産関数とは名前の通り、代替の弾力性が一定の関数です。生産量をY、資本をK、労働力をLとした場合に、

Y = (\alpha K^{\sigma}+(1-\alpha)L^{\sigma})^{\frac{1}{\sigma}}

で表される生産関数のことです。$0 \le \alpha \le 1$であり、$\sigma$の値によっては「ces型」を「線形」・「コブダグラス型」・「レオンチェフ型」に変形させることができます。この記事ではこの三つの変形に関することが主となります。

ces型についての詳しい解説は「コブダグラス関数、CES関数 - 経済学メモ - アットウィキ」、「ぴーすけ講座 CES関数について - わかりやすい経済学」を参考にしてください。
テキストに関しては以下を参照しています。
・ミクロ経済学　著:西村和雄
・Microeconomics Analysis 著:Hal R. Varian

コードの実装

早速コードの実装を行います。結果だけ見たい人は読み飛ばしていただいて構いません。
コードの実装にあたり「jupyter notebookで触れるプロットを描く」を参考にさせていただきました。

全体のコード

ここでお詫びしなければならないことがあります。それはプログラム内で対数をとったりロピタルの定理を使って変換を行っていないことです。if文を使って$\sigma$の値がある値の時にある型に変化するよう書いていますので厳密性を欠いています(?)。また近似的に表示をさせることができるプログラムなので厳密なものを書ける方はぜひよろしくお願いします。
それではコードです。


%matplotlib notebook
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import numpy as np


a = float(input("aの値を入力してね"))
sigma = float(input("sigmaの値を入力してね")) 

# sigma = 1 の時　線形
# sigma = 0 の時　コブダグラス型
# sigma = -∞の時(-100や-200で近似的に表示)　レオンチェフ型

if sigma == 0:
    def f(k,l):
        return (k**a) * (l**(1-a))

else:
    def f(k,l):
        return (a*k**(sigma) + (1-a)*l**(sigma))**(1/sigma)


k = np.arange(0, 100, 1)
l = np.arange(0, 100, 1)

K, L = np.meshgrid(k, l)
Y = f(K,L)

fig = plt.figure()
ax = Axes3D(fig)

ax.set_xlabel("K")
ax.set_ylabel("L")
ax.set_zlabel("Y")

ax.plot_wireframe(K, L, Y)
#ax.contour(K, L, Y,20)  #等高線ver, 最後の数字は線の数

plt.show()