![](/img/trans.png)
[英]Selecting rows based on multiple column values in pandas dataframe MultiIndex
[英]Selecting rows based on sum over multiindex in Pandas
import pandas as pd
import numpy as np
np.random.seed(365)
rows = 100
data = {'Month': np.random.choice(['2014-01', '2014-02', '2014-03', '2014-04'], size=rows),
'Code': np.random.choice(['A', 'B', 'C'], size=rows),
'ColA': np.random.randint(5, 125, size=rows),
'ColB': np.random.randint(0, 51, size=rows),}
df = pd.DataFrame(data)
df = df[((~((df.Code=='A')&(df.Month=='2014-04')))&(~((df.Code=='C')&(df.Month=='2014-03'))))]
dfg = df.groupby(['Code', 'Month']).sum()
上面給出了我的 dataframe。我只想 select 那些總和(ColA)超過 1000 的條目,當這個總和在級別 [0] 上執行時
dfg.ColA.sum(level=[0])
dfg[dfg.ColA.sum(level=[0])>1000]
以上一個拋出錯誤? 預計 output 是:
ColA ColB
Code Month
B 2014-01 477 300
2014-02 591 167
2014-03 522 192
2014-04 367 169
C 2014-01 412 180
2014-02 275 205
2014-04 901 309
您需要使用groupby
+ transform
在level=0
索引中廣播總和值
dfg[dfg.groupby(level=0)['ColA'].transform('sum').gt(1000)]
ColA ColB
Code Month
B 2014-01 477 300
2014-02 591 167
2014-03 522 192
2014-04 367 169
C 2014-01 412 180
2014-02 275 205
2014-04 901 309
另一種方法做同樣的事情:
groups = [g for _,g in df.groupby('Code') if g.ColA.sum()>1000]
pd.concat(groups).groupby(['Code', 'Month']).sum()
'''
ColA ColB
Code Month
B 2014-01 477 300
2014-02 591 167
2014-03 522 192
2014-04 367 169
C 2014-01 412 180
2014-02 275 205
2014-04 901 309
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.