使用 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

使用 Python 中的奇数创建行数

Create row count using odd numbers in Python

python

pandas

group-by

count

row