在 Pandas Groupby 中使用 Column 和 Row Multi-Index 值而不取消堆叠

Question

我有一个多索引层次结构设置如下：

import numpy as np

sectors = ['A','B','C','D']
ports = ['pf','bm']
dates = range(1,11)*2
wts, pchg = zip(*np.random.randn(20,2))
df = pd.DataFrame(dict(dates=dates,port=sorted(ports*10),
                       sector=np.random.choice(sectors,20), wts=wts,
                       pchg=pchg))

df = df.set_index(['port','sector','dates'])
df = df.unstack('port')
df = df.fillna(0)

我想按 dates 和 port 分组，然后求和 pchg * wts

我已经阅读了文档，但我正在努力解决这个问题。

非常感谢任何帮助。谢谢

Answer 1

你确实不需要拆开来得到你想要的东西，使用product方法来做你想要的乘法。一步一步：

从此数据帧开始：

In [50]: df.head()
Out[50]:
                  pchg                 wts
port                bm        pf        bm        pf
sector dates
A      1      0.138996  0.451688  0.763287 -1.863401
       3      1.081863  0.000000  0.956807  0.000000
       4      0.207065  0.000000 -0.663175  0.000000
       5      0.258293 -0.868822  0.109336 -0.784900
       6     -1.016700  0.900241 -0.054077 -1.253191

我们可以先用product方法做pchg * wts部分，乘以轴1，但只适用于第二层：

In [51]: df.product(axis=1, level=1).head()
Out[51]:
port                bm        pf
sector dates
A      1      0.106094 -0.841675
       3      1.035134  0.000000
       4     -0.137320  0.000000
       5      0.028241  0.681938
       6      0.054980 -1.128174

然后我们可以按 dates 分组（不再需要按端口分组）并求和：

In [52]: df.product(axis=1, level=1).groupby(level='dates').sum()
Out[52]:
port         bm        pf
dates
1      0.106094 -0.841675
2      0.024968  1.357746
3      1.035134  1.776464
4     -0.137320  0.392312
5      0.028241  0.681938
6      0.054980 -1.128174
7      0.140183 -0.338828
8      1.296028 -1.526065
9     -0.213989  0.469104
10     0.058369 -0.006564

这给出了与

相同的输出

df.stack('port').groupby(level=[1,2]).apply(lambda x: (x['wts']*x["pchg"]).sum()).unstack('port')

在 Pandas Groupby 中使用 Column 和 Row Multi-Index 值而不取消堆叠

Use Column and Row Multi-Index values in Pandas Groupby without unstacking

python

pandas

multi-index