python - 添加一个填充列的新列，对于前 6 行，用 1 到 6 个数字填充它，这与使用 python 的下 6 行相同

Question

我正在尝试添加一个新列，其中数据框中的每 6 行填充 1 到 6 个数字。对数据框中的所有行重复此操作。下图显示了输出的样子

输入

    ID
0   20
1   20
2   20
3   20
4   20
5   20
6   34
7   34
8   34
9   34
10  34
11  34
12  67
13  67
14  67
15  67
16  67
17  67

输出

    ID  6_months
0   20         1
1   20         2
2   20         3
3   20         4
4   20         5
5   20         6
6   34         1
7   34         2
8   34         3
9   34         4
10  34         5
11  34         6
12  67         1
13  67         2
14  67         3
15  67         4
16  67         5
17  67         6

标签： pythonpreprocessor