Pandas:如何创建一个简单的计数器，每增加n行

Question

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

示例===>计数器每增加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]