pythonでグラフ描く度に調べてるので、最もシンプルな使い方だけまとめとく

jupyter notebook使ってます。
jupyter notebookじゃないのなら「%matplotlib inline」は不要

２次元グラフ

# ライブラリのインポート
%matplotlib inline
import matplotlib.pyplot as plt
import numpy as np

# グラフ描画したい関数
def func(x):
    return x**2

# 変数準備
# np.arange(開始, 終了, ステップ)
x = np.arange(0.0, 10.0, 0.1)
y = func(x)

# グラフに描画
plt.plot(x, y);

３次元グラフ

# ライブラリのインポート
%matplotlib inline
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import numpy as np

# グラフ描画したい関数
def func(x, y):
    return x * y * (10 - x - y)

# 変数準備
x = np.arange(0.0, 10.0, 0.1)
y = np.arange(0.0, 10.0, 0.1)
X, Y = np.meshgrid(x, y)
Z = func(X, Y)

# 描画
fig = plt.figure()
ax = Axes3D(fig)
ax.set_xlabel("x")
ax.set_ylabel("y")
ax.set_zlabel("f(x, y)")
ax.plot_wireframe(X, Y, Z)
plt.show()