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

时间:2019-11-01 04:34:31

标签: python pandas

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行”并分配1,然后重复然后扩展“ myVar行”并递增到2,依此类推。

所需的输出:

enter image description here

1 个答案:

答案 0 :(得分: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