以下はChatGPTの操作例です。怖いぐらいに凄いので細かいことは書きません。
以下の2つの質問をするだけで、たまにしか使わないから、忘れがちなpandasのgroupbyを含むfunctionを豊富な解説付きで作ってくれます。
質問した内容
- 以下のデータで月を入力としてccode,volumeの月ごとの中央値を返すルーチンを作って+数十行のcsvサンプル
- 上のプログラムをサブルーチン化して
ChatGPTの作ったfunction
import pandas as pd
def calc_median_volume_by_month(df, month):
# datetime型の列を作成
df['date'] = pd.to_datetime(df['date'])
# 指定された月のデータのみを抽出
df_month = df[df['date'].dt.month == month]
# ccodeとvolumeのみを抽出
df_month_volume = df_month[['ccode', 'volume']]
# ccodeでグループ化し、中央値を計算
median_volume_by_ccode = df_month_volume.groupby('ccode').median()
return median_volume_by_ccode