如何求和索引大於X的所有值？

Question

比方說，我有這個系列：

>>> s = pd.Series({1:10,2:5,3:8,4:12,5:7,6:3}) 
>>> s 
1 10 
2  5 
3  8 
4 12 
5  7 
6  3 

我要總結的所有值該指數大於X.因此，如果如X = 3，我想這一點：

>>> X = 3 
>>> s.some_magic(X) 
1 10 
2  5 
3  8 
>3 22 

我設法做到這一點在這個相當笨拙的方法：

lt = s[s.index.values <= 3] 
gt = s[s.index.values > 3] 
gt_s = pd.Series({'>3':sum(gt)}) 
lt.append(gt_s) 

，並得到想要的結果，但我認爲應該有一種更簡單，更優雅的方式......還是有？

Answer 1

s.groupby(np.where(s.index > 3, '>3', s.index)).sum() 

或者，

s.groupby(s.index.to_series().mask(s.index > 3, '>3')).sum() 
Out: 
1  10 
2  5 
3  8 
>3 22 
dtype: int64 

Answer 2

這裏是一個可能的解決方案：

import pandas as pd 

s = pd.Series({1: 10, 2: 5, 3: 8, 4: 12, 5: 7, 6: 3}) 
iv = s.index.values 

print s[iv <= 3].append(pd.Series({'>3': s[iv > 3].sum()}))