Python学习笔记：季度判断

一、使用datetime.quarter判断

# Python月份判断季度
import datetime
import pandas as pd
df['quarter'] = df['date'].dt.quarter

二、字典手动判断

data = pd.DataFrame({'month':[3,6,9]})

quarter = {"春":[3,4,5],
      "夏":[6,7,8],
      "秋":[9,10,11],
      "冬":[12,1,2]}
      
def f(x):
    for k, v in quarter.items():
        if x in v:
            return k
            
data['quarter'] = data['month'].apply(f)
print(data)
'''
   month quarter
0      3       春
1      6       夏
2      9       秋
'''