How add k-fold no in a column?

Question

I have a pandas data frame and I want to create a column "5_fold" and in that column, it should display which row is belonging to which fold.

I tried:

<