等高線したい(matplotlib.axes.Axes.contour)
1import numpy as np
2import matplotlib.pyplot as plt
3
4# データを準備する
5x = np.linspace(0, 10, 100) # 0から10までの100点を生成
6y = np.linspace(0, 10, 100) # 0から10までの100点を生成
7X, Y = np.meshgrid(x, y) # グリッドを作成
8Z = np.sin(X) * np.cos(Y) # ZはXとYの関数(例:sin(X)*cos(Y))
9data = Z # 2次元配列
10
11# キャンバスを作成
12fig, ax = plt.subplots()
13
14# 等高線を作成
15contour = ax.contour(
16 X, Y, data, # X、Y、Zの値を指定
17 levels=10, # 等高線のレベル数
18 cmap="viridis", # カラーマップ
19)
20
21# カラーバーを追加
22fig.colorbar(contour, ax=ax)
23plt.show()
Axes.contourで等高線を作成できます。
X、Y、Zの値を指定し、オプションで等高線のレベル数やカラーマップを変更できます。