基于索引拆分DataFrame

Question

我有一个DataFrame，我想根据索引的字符串属性将它拆分为三个DataFrames。索引由ID组成，前两个字母表示国家/地区，例如

DE1
UK4
US5
DE2
UK1
US3

我想要三个DataFrames，索引是：

DE1
DE2

UK1
UK4

US3
US5

这看起来很有希望：

df.groupby(df.index.str[:2]).groups

但我不知道如何用它来解决我的问题。

Answer 1

这里有一种方法可以做到

split = []
for value in df.index.str[:2].unique().values:
    split.append(df[df.index.str[:2] == value])

我们首先使用数据框中的前2个字母计算唯一的国家/地区代码。然后我们循环它们，并使用所有唯一的国家/地区代码索引到数据帧中。在这里，我只是将结果DataFrame添加到一个数组中。