将结果从groupby移动到DataFrame

Question

我使用此方法来求和我数据框中某些条件的计数

 combinations = combinations.groupby(['COUNTRY', 'COLORS']).size().unstack(fill_value=0)

这说明了我正在尝试做的事情：Dataframe containing counts where column conditions are met

即一个看起来像这样的数据框：

   COUNTRY         BLUE        YELLOW     COLOR...
0  England         1232        3312       Sum
1  Hungary         5543        7661       Sum
2  ETC             ETC         ETC

但是对我来说，结果数据帧看起来像这样

      COLORS       BLUE        YELLOW     COLOR...
   COUNTRY         
0  England         1232        3312       Sum
1  Hungary         5543        7661       Sum
2  ETC             ETC         ETC

所以我有想要的值，但是我似乎无法遍历所有结果。尝试从groupby数据框中访问列时，它将引发KeyError。

colors = combinations.loc['COLORS']

KeyError: 'the label [COLORS] is not in the [index]'

我有一个包含国家纬度/经度的数据框，我想将groupby的结果附加到该数据框。

因此，我不想完全使用COLORS列，而是希望将这些布局与这些列一起使用：  Country lat lon BLUE YELLOW RED ETC...

其中每种颜色都具有与上一个数据帧不同的颜色值。

这是否可能与熊猫有关，还是应该以某种特殊的方式遍历结果数据帧并将它们逐个添加到新的数据帧中？

让我知道是否应该澄清一些事情。

感谢您的帮助