python - 将迭代数分配给变量的 n 行

Question

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

电流输入：

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

如何使用变量 (myVar = 4) 枚举每个 'myVar rows' 并赋值 1，然后重复 ext 'myVar rows' 并递增到 2，依此类推。

期望的输出：

标签： pythonpandas

Answer 1

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

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