Python 从变量为n行分配迭代数

电流输入：

import pandas as pd
ndf = pd.DataFrame({'col1':range(6)})
print(ndf)

如何使用变量myVar=4枚举每个“myVar行”并赋值1，然后重复ext“myVar行”并递增到2，依此类推

期望输出：

---

看起来你可以用一点整数除法来解决这个问题：

   col1
0     0
1     1
2     2
3     3
4     4
5     5

---

你是说行只能被分配1或2？不，这是一个例子。如果我有22行，所以不总是平均分组（或100行），那么前4行将被分配为“1”，接下来的4行将被分配为“2”，依此类推。

df['subgroup'] = df.index // myVar + 1
df
    col1  subgroup
0      0         1
1      1         1
2      2         1
3      3         1
4      4         2
5      5         2