在 Pandas groupby 的情节中保持秩序

qfd 发表于 Dev

qfd

我有一个如下所示的数据框

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 排序。

谢谢你的帮助！

亚历山大

问题在于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()):
    ...

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

本文收集自互联网，转载请注明来源。

如有侵权，请联系 [email protected] 删除。