pandasのindex番号に繰り返した連番を振りたい

Question

前提・実現したいこと

pandasのindex番号に繰り返した連番を振りたいです。上から順に0〜ではなく、0〜4を最後の行まで繰り返したいです。
reset_indexを試してみたのですが、これでは上から順に番号が振られるだけなのでやりたいことができませんでした。
ご教授のほどよろしくお願いいたします。

                 datetime
0     2021-10-01 00:00:00
1     2021-10-01 00:30:00
2     2021-10-01 01:00:00
3     2021-10-01 01:30:00
4     2021-10-01 02:00:00
...                   ...
0  　　　　 2021-11-23 03:30:00
1  　　　　　　2021-11-23 04:00:00
2  　　　　　　2021-11-23 04:30:00
3  　　　　　　2021-11-23 05:00:00
4  　　　　　　2021-11-23 05:30:00

該当のソースコード

                 datetime
0     2021-10-01 00:00:00
1     2021-10-01 00:30:00
2     2021-10-01 01:00:00
3     2021-10-01 01:30:00
4     2021-10-01 02:00:00
...                   ...
2552  2021-11-23 03:30:00
2553  2021-11-23 04:00:00
2554  2021-11-23 04:30:00
2555  2021-11-23 05:00:00
2556  2021-11-23 05:30:00

Answer 1

df.index = df.index % 5

でできます。

実行結果

>>> print(df)
              datetime
0  2021-10-01 00:00:00
1  2021-10-01 00:30:00
2  2021-10-01 01:00:00
3  2021-10-01 01:30:00
4  2021-10-01 02:00:00
5  2021-10-01 02:30:00
6  2021-10-01 03:00:00
7  2021-10-01 03:30:00
8  2021-10-01 04:00:00
9  2021-10-01 04:30:00
10 2021-10-01 05:00:00
>>> df.index = df.index % 5
>>> print(df)
             datetime
0 2021-10-01 00:00:00
1 2021-10-01 00:30:00
2 2021-10-01 01:00:00
3 2021-10-01 01:30:00
4 2021-10-01 02:00:00
0 2021-10-01 02:30:00
1 2021-10-01 03:00:00
2 2021-10-01 03:30:00
3 2021-10-01 04:00:00
4 2021-10-01 04:30:00
0 2021-10-01 05:00:00

ただし、インデックスやカラムに重複があると、locやatがうまく動かなかったりなどの問題が発生するので、インデックスを変更するよりも、やりたい処理の中でdf.index % 5のような式を使う方が良い場合が多いです。

Comments

[yama0309]

こんなスマートなやり方でできるとは感動しました。無事解決しました、ありがとうございます。
下の方のご回答のようにリスト作成も試したのですが、行数が多い＆行が増えていくので、その対処もできないかと思っておりました。
お二方ともありがとうございました。

Answer 2

セットしたい値を持つリストを作成してindexに代入すればよいです。

import pandas as pd

df = pd.DataFrame({'datetime':pd.date_range('2021-10-01 00:00:00', '2021-10-01 05:00:00', freq='30min')})

row = len(df)
index = (list(range(5))*(row//5+1))[:row]
print(index) # [0, 1, 2, 3, 4, 0, 1, 2, 3, 4, 0]
df.index = index
print(df)
"""
             datetime
0 2021-10-01 00:00:00
1 2021-10-01 00:30:00
2 2021-10-01 01:00:00
3 2021-10-01 01:30:00
4 2021-10-01 02:00:00
0 2021-10-01 02:30:00
1 2021-10-01 03:00:00
2 2021-10-01 03:30:00
3 2021-10-01 04:00:00
4 2021-10-01 04:30:00
0 2021-10-01 05:00:00
"""