【python】コンター図の描画【ChatGPT】

以下は、(r, θ, z)の形式のデータファイルから等高線図を描くPythonプログラムです。プログラムでは、pandasを使用してデータファイルを読み込み、matplotlibを使用して等高線図を作成しています。

import pandas as pd
import matplotlib.pyplot as plt
import numpy as np

# データファイルを読み込む
df = pd.read_csv('data.csv', header=None, names=['r', 'theta', 'z'])

# r, theta, zの値を配列に変換する
r = np.array(df['r'])
theta = np.array(df['theta'])
z = np.array(df['z'])

# r, theta, zの値からグリッドデータを生成する
r_grid, theta_grid = np.meshgrid(r, theta)
z_grid = np.reshape(z, (len(theta), len(r)))

# 等高線図を作成する
plt.contourf(r_grid * np.cos(theta_grid), r_grid * np.sin(theta_grid), z_grid, cmap='coolwarm')
plt.colorbar()
plt.show()

以前、コンター図の描画について書いた記事
https://sciencompass.com/machine-learning/python_contour