在Python中通过标识符子集成几个数据框？

Question

我希望通过标识符在Python中对数据框进行子集化。例如，假设我们有以下数据：

ID    Number
A     50
A     45
A     21
B     78
B     79
B     12
C     15
C     74
C     10

我想将数据分成三个独立的数据帧，即A的所有数据都是第一个数据帧，B是第二个，C是第三个。

我无法解决这个问题。我尝试将设置用于唯一值，但我认为这不是解决问题的方法。任何帮助表示赞赏。

Answer 1

这是你想要的吗？ （PS：我认为自动指定名称为Dataframe）

variables = locals()
for i in df['ID'].unique():
    variables["df{0}".format(i)] = df.loc[df.ID == i,]

dfA
Out[147]: 
  ID  Number
0  A       1
3  A       1
6  A       1