[問題]groupby 做邏輯判斷？

[gene50814 (genechen)]

各位大大好
小弟有個時間序列的資料
範例如下：
https://i.imgur.com/1Z6Or19.jpg

小弟想做的事是要算每個月各公司市值的中位數 若公司大於等於中位數則新的column給
值’big’，反之則’small’

想完成的感覺如下圖：
https://i.imgur.com/1urJNVp.jpg

小弟想應該是要先把date groupby起來，但不知道該如何做後續的邏輯判斷

請各位大大指教了 謝謝

--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 111.240.146.107 (臺灣)
※ 文章網址: https://www.ptt.cc/bbs/Python/M.1588495612.A.C6E.html
推 yuasa: 你用pandas嗎?這兩個表除了多了B/S欄差在哪?你的B/S決定的
→ yuasa: 條件是?
推 TitanEric: 樓上你沒看內文齁
→ gene50814: 是用pandas的  後來我是用join把groupby之後的series加
→ gene50814: 到原本的df再做邏輯判斷  不知道有沒有更好的作法？
推 moodoa3583: 如果是用data[B/S] = data[MarketValue].apply(lambda
→ moodoa3583:  x:True if x >np.median(data[MarketValue]) else Fa
→ moodoa3583: lse) 呢?
→ moodoa3583: *>=
推 lycantrope: for date, tempdf in data.groupby("date"):
→ lycantrope:  BS = tempdf["Price"] > tempdf["Price"].median()
→ lycantrope:     df.loc[BS.index,"B/S"] = BS
→ lycantrope: 打錯df改成date     data.loc[BS.index,"BS"] = BS
→ lycantrope: 把groupby 當iter for loop 判斷邏輯直接輸入data