使用Python中的奇数创建行计数

Question

这里必须有一个简单的解决方案。我知道如何使用cumcount，但我希望它数数奇数。例如，我有以下DF。搜索非常毫无用处，因为它们都在计算“奇数”数字。

letters = [ "A", "A", "A", "B", "B", "B"]

df = pd.DataFrame(letters, columns=["letter"])

df['cumcount'] = df.groupby('letter').cumcount() + 1


df =

letter     cumcount  

  A           1
  A           2
  A           3
  B           1
  B           2
  B           3

我想做的就是输出这一点。

df =

letter     cumcount    odd_count

  A           1            1
  A           2            3
  A           3            5
  B           1            1
  B           2            3
  B           3            5

Answer 1

让我们做一些数学：

df['cumcount'] = df.groupby('letter').cumcount() * 2 + 1

---

  letter  cumcount
0      A         1
1      A         3
2      A         5
3      B         1
4      B         3
5      B         5