python - 熊猫，根据其他列创建“订单”列

Question

我有两列数据框，集群标题和它们所属的章节。我想创建第三列，包含该集群在章节中的“顺序”或位置。

所以，我想打开以下数据框：

cluster_title, chapter
"rabbits",   1
"horses",    1
"cows",      1
"trains",    2
"airplanes", 2
"ships",     2
"carrot",    3
"potato",    3
"tomato",    3

变成这样：

cluster_title, chapter, position_in_chapter,
"rabbits",   1, 1
"horses"     1, 2
"cows",      1, 3
"trains",    2, 1
"airplanes", 2, 2
"ships",     2, 3
"carrot",    3, 1
"potato",    3, 2
"tomato",    3, 3

我尝试使用group_by函数来接近它并以某种方式使用索引，但要么我遗漏了一些明显的东西（很可能），要么这是错误的方法，因为生成的对象需要额外的步骤，这似乎把我带到了错误的方向。

有人能指出我正确的方向吗？

标签： pythonpandas

Answer 1

尝试使用groupby和cumcount：

df["position_in_chapter"] = df.groupby("chapter").cumcount()+1

>>> df
  cluster_title  chapter  position_in_chapter
0       rabbits        1                    1
1        horses        1                    2
2          cows        1                    3
3        trains        2                    1
4     airplanes        2                    2
5         ships        2                    3
6        carrot        3                    1
7        potato        3                    2
8        tomato        3                    3