创建一个连续计数4行和6行的pandas列计数器

Question

我有一个包含多列的 pandas 数据框，我想创建一个连续计数 4 行和 6 行的计数器。 我希望它看起来像下面的数据框：

index	counter
0	1
1	1
2	1
3	1
4	2
5	2
6	2
7	2
8	2
9	2
10	3
11	3
12	3
13	3

如您所见，counter 列中的前 4 个值是 1，然后接下来的 6 个值是 2，然后再接下来的 4 个值是 3。

Answer 1

您的问题在编辑后有点清楚，您可以创建一个空列表和一个计数器变量，然后迭代行数范围，将其递增 10 即（4+6），然后在每次迭代时，创建所需的长度为 4 和 6 的列表，带有 counter 和 counter+1，将其添加到结果列表中。 最后从结果列表中取出切片的第一个 df.shape[0] 值（因为它的值可能比 df.shape[0] 多一些），并将其分配给新列 df['counter']。

result = []
counter = 1
for i in range(0, df.shape[0], 10):
    result += [counter]*4
    counter += 1
    result += [counter]*6
    counter += 1

df['counter'] = result[:df.shape[0]]
    
# result [1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 3, 3, 3, 3, 4, 4, 4, 4, 4, 4]

输出：

    index  counter
0       0        1
1       1        1
2       2        1
3       3        1
4       4        2
5       5        2
6       6        2
7       7        2
8       8        2
9       9        2
10     10        3
11     11        3
12     12        3
13     13        3