python groupbyによる計算方法

Question

groupbyを使って、うまく計算する方法を探しています。
以下のようなdataframeがあるとき、dateとidでグループを作り、そのグループ内で計算を行いたいと思っています。

元のdataframe
date, id, col_1, col_2
2020-04-01, 1, a1, a2
2020-05-01, 1, b1, b2
2020-06-01, 1, c1, c2
2020-04-01, 2, A1, A2
2020-05-01, 2, B1, B2
2020-06-01, 2, C1, C2

取得したい結果
2020-04-01, (a10.6 + A10.4 の値), (a20.3 + A20.7 の値)

groupbyでsumやmaxなどではなく、少し複雑な計算をさせる場合はどのようにすればよいのでしょうか？
ご教示いただけますと幸いです。
よろしくお願い致します。

Answer 1

dateとidでグループを作り

というよりdateでグループ化すべきなのでは？
そのうえでidが２種類しかないなら以下のような感じでできます。

import pandas as pd
import numpy as np

df = pd.DataFrame({'date':[4,5,6,4,5,6],'id':[1,1,1,2,2,2], 'col_1':[10,20,30,100,200,300], 'col_2':[40,50,60,400,500,600]})
print(df)
"""
   date  id  col_1  col_2
0     4   1     10     40
1     5   1     20     50
2     6   1     30     60
3     4   2    100    400
4     5   2    200    500
5     6   2    300    600
"""

grp = df.groupby(['date'])
for date, dfg in grp:
    c1 = np.dot( dfg['col_1'], [0.6, 0.4])
    c2 = np.dot( dfg['col_2'], [0.3, 0.7])
    print(date, c1, c2)
#4 46.0 292.0
#5 92.0 365.0
#6 138.0 438.0

Comments

[ta.eee]

ありがとうございました！
dateとidでグループ化しようとして、なかなかうまくいきませんでしたが、dateだけで良かったのですね。