python月份判断季度

Question

data["max"]=data.eq(data.max(1),axis=0).dot(data.columns)

data["max"]    #这个输出是月份

jd={"春":[3,4,5],"夏":[6,7,8],"秋":[9,10,11],"冬":[12,1,2]}

用这两个数据判断季度

Answer 1

用apply函数，这样操作：

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

# 这个输出是月份

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

#用这两个数据判断季度
data['jd']=data['max'].apply(f)
print(data)

如果有帮助请点一下我回答右上方的采纳，谢谢