Question

我有以下数据：

   import pandas as pd
x4 = pd.DataFrame({"ID": [101,101, 102, 103, 104, 105],
                   "Prob": [1, 1,1, 1, 1, 1],
                   "Ef": [0,2, 0, 0, 0.25, 0.29],
                   "W": [2, 2,3, 4, 5, 6],
                   "EC": [0, 0,0, 0, 1.6, 2],
                   "Rand": [11, 12,12, 13, 14, 15]})

我想获取sum(Prob * Ef) by ID ，然后只保留列ID，列sum，{{ 1}}列和EC列。

所以最后我想要这个：

我试过这个：ID sum_column EC W 1: 101 2.00 0.0 2 2: 101 2.00 0.0 2 3: 102 0.00 0.0 3 4: 103 0.00 0.0 4 5: 104 0.25 1.6 5 6: 105 0.29 2.0 6

但它不起作用

Answer 1

乘以GroupBy.transform乘以列：

class Myplayer: AVPlayer{
    static var sharedInstance: Myplayer = Myplayer()
    override private init() {
            super.init()
            // Your code for init object
        }
    }

如果列的顺序很重要，请使用insert：

x4['sum_column'] = x4['Prob'].mul(x4['Ef']).groupby(x4['ID']).transform('sum')
x4 = x4.drop(['Ef','Prob', 'Rand'], axis=1)
print (x4)
    ID  W   EC  sum_column
0  101  2  0.0        2.00
1  101  2  0.0        2.00
2  102  3  0.0        0.00
3  103  4  0.0        0.00
4  104  5  1.6        0.25
5  105  6  2.0        0.29

groupby和用于某些列的pandas的总和，同时还包括其他列

1 个答案: