【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

sciencompass34 has written 159 articles

はじめまして!”あおやぎ”と言います。
メーカーで研究開発の仕事をしています。このブログでは、私の専門分野である半導体やそれに関連する内容を紹介していきます。
半導体関連の知識をまとめたデータベースのようにしたいなと思っています。

Leave a Reply

Your email address will not be published. Required fields are marked *

You may use these HTML tags and attributes: <a href="" title=""> <abbr title=""> <acronym title=""> <b> <blockquote cite=""> <cite> <code> <del datetime=""> <em> <i> <q cite=""> <s> <strike> <strong>

このサイトはスパムを低減するために Akismet を使っています。コメントデータの処理方法の詳細はこちらをご覧ください