Pandasを使って旬をもとめる

Question

前提・実現したいこと

csvファイルの日付欄から旬を求めたいです。

発生している問題・エラーメッセージ

日付欄には"2019-11-3"と言う風に日付が入っています。
そこから月のみは抽出したのですが、そこから旬を求めることができませんでした。
if文を使ってみたのですが、すべて同じ季節になってしまい、うまくできませんでした。
そして階層型インデックスを使ってみてそれはできたのですが、他の方法があれば知りたいです。

該当のソースコード

lunch_box["datetime"] = pd.to_datetime(lunch_box["datetime"])

# 月の抽出
lunch_box["month"] = lunch_box["datetime"].dt.month

lunch_box.loc[lunch_box['month'] == 3, 'season'] = '春'
lunch_box.loc[lunch_box['month'] == 4, 'season'] = '春'
lunch_box.loc[lunch_box['month'] == 5, 'season'] = '春'
lunch_box.loc[lunch_box['month'] == 6, 'season'] = '夏'
lunch_box.loc[lunch_box['month'] == 7, 'season'] = '夏'
lunch_box.loc[lunch_box['month'] == 8, 'season'] = '夏'
lunch_box.loc[lunch_box['month'] == 9, 'season'] = '秋'
lunch_box.loc[lunch_box['month'] == 10, 'season'] = '秋'
lunch_box.loc[lunch_box['month'] == 11, 'season'] = '秋'
lunch_box.loc[lunch_box['month'] == 12, 'season'] = '冬'
lunch_box.loc[lunch_box['month'] == 1, 'season'] = '冬'
lunch_box.loc[lunch_box['month'] == 2, 'season'] = '冬'

補足情報（FW/ツールのバージョンなど）

上のコードも改善できるとは思いますが、今は他の方法で”旬”を出す方法があれば知りたいです。

Answer 1

pd.cut(df["date"].dt.month % 12, [0, 3, 6, 9, 12], labels=["冬", "春", "夏", "秋"], right=False)

Answer 2

整数 (1 ~ 12) -> 季節の文字列という変換関数を用意して、Series.map() で適用すればよいです。

import pandas as pd

df = pd.DataFrame(
    {"date": pd.date_range(start="2020/01/01", end="2020/12/31", freq="M")}
)


def to_season(m):
    if 1 <= m <= 3:
        return "春"
    if 4 <= m <= 6:
        return "夏"
    if 7 <= m <= 9:
        return "秋"
    if 10 <= m <= 12:
        return "冬"


df["season"] = df["date"].dt.month.map(to_season)
print(df)
#          date season
# 0  2020-01-31      春
# 1  2020-02-29      春
# 2  2020-03-31      春
# 3  2020-04-30      夏
# 4  2020-05-31      夏
# 5  2020-06-30      夏
# 6  2020-07-31      秋
# 7  2020-08-31      秋
# 8  2020-09-30      秋
# 9  2020-10-31      冬
# 10 2020-11-30      冬
# 11 2020-12-31      冬

Comments

[退会済みユーザー]

ありがとうございます。
解決することができました。