Question

我的数据框如下所示

univ  date            ms      cs
 C    11/01/1977      0.2     0.1
 C    11/02/1977      0.3     0.4
 B    11/01/1977      0.6     -0.1
 B    11/02/1977      0.55    -0.5
 A    11/01/1977      0.23    0.2
 A    11/02/1977      0.12    0.15
 etc

这就是我目前所拥有的

fig, axes = plt.subplots(1, 3, figsize=(15, 5))

for (universe, group), ax in zip(df.groupby([df.univ]), axes.flatten()):
    group.plot(x='ms', y='cs', kind='scatter', ax=ax, title=universe)
    ax.set_xlim(0, 1)
    ax.set_ylim(0, 1)

这会生成3个散点图，限制范围为0到1。我想做的是保持标题为C，B，A的图的顺序，使其按A，B，C排序。

感谢您的帮助！

Answer 1

问题在于groupby默认情况下以排序顺序返回项目。只需设置标记sort=False。

for (universe, group), ax in zip(df.groupby([df.univ], sort=False), axes.flatten()):
    group.plot(x='ms', y='cs', kind='scatter', ax=ax, title=universe)
    ax.set_xlim(0, 1)
    ax.set_ylim(0, 1)

您也可以按照您想要的任何顺序返回它们。

gb = df.groupby('univ')
desired_order = ('B', 'A', 'C')
for (universe, group), ax in zip(((u, gb.get_group(u)) for u in desired_order), 
                                 axes.flatten()):
    ...

顺便说一下，每个帖子你应该只有一个问题。

在pandas groupby的情节中保持秩序

1 个答案: