python - Pandas:如何创建一个简单的计数器来增加每 n 行？

Question

有没有办法创建一个每 n 行增加 1 的计数器？

示例 ===> 计数器每 4 行递增:

        counter
    0    1
    1    1
    2    1
    3    1
    4    2
    5    2
    6    2
    7    2
    8    3
    9    3

我正在尝试 df['counter'] = np.arange(len(df)//4) 但我得到了索引错误的长度

谢谢

Answer 1

你很接近，需要:

df['counter'] = np.arange(len(df)) // 4 + 1
print (df)
   counter
0        1
1        1
2        1
3        1
4        2
5        2
6        2
7        2
8        3
9        3

您的解决方案不起作用，因为具有 2 个值的数组已分配给列:

print (len(df)//4)
2

print (np.arange(len(df)//4))
[0 1]

但是如果需要像 DataFrame 这样大小相同的数组:

print (np.arange(len(df)))
[0 1 2 3 4 5 6 7 8 9]