python - 在 Pandas 中找到同名的群组？

我有两个DataFrame，我按使用月份将它们分成几组：

grouped1 = data1[key].set_index('Time').groupby(pd.TimeGrouper(freq='M'))
grouped2 = data2[key].set_index('Time').groupby(pd.TimeGrouper(freq='M'))

grouped2的组可能比grouped1多，反之亦然。我想在分组的DataFrame之间共享相同名称的组上运行统计信息。

例如，如果grouped1有一个名为April的组，而grouped2有一个名为April的组，我想在这些组上运行统计信息。我想遍历所有使用相同名称的群组。

我该怎么做？
`

最佳答案

您可以使用groupby从.groups对象获取组，这将返回一个dict，键为组值，然后可以使用set和intersection找到公共列，然后使用：

In [252]:
df = pd.DataFrame({'id':['a','b','c','b','c'], 'data':np.random.randn(5)})
df1 = pd.DataFrame({'id':['a','a','a','b','d'], 'data':np.random.randn(5)})
gp1 = df.groupby('id')
gp2 = df1.groupby('id')
print(gp1.groups)
print(gp2.groups)

{'b': [1, 3], 'a': [0], 'c': [2, 4]}
{'b': [3], 'a': [0, 1, 2], 'd': [4]}

In [ ]:
common = set(gp1.groups.keys()).intersection(set(gp2.groups.keys()))
common

In [270]:
for gp in common:
    print('first:\n')
    print(gp1.get_group(gp).head())
    print('\nsecond: \n')
    print(gp2.get_group(gp).head())

first:

       data id
1 -0.252577  b
3  0.274945  b

second:

       data id
3  1.521554  b
first:

       data id
0  0.142474  a

second:

       data id
0  1.536180  a
1 -1.117422  a
2  1.490307  a