dropメソッドを用いて、変数以降の全ての行の削除

Question

前提・実現したいこと

指定したインデックス以降の行にあるデータを削除し、その保存を行いたいです。
また、それを１つずつインデックスを増やしながら行いたいです。
そこでfor文を用いて、dropの中に変数を入れてみたのですが、上手くいきません。
間違いがわかる方がいらっしゃいましたら、よろしくお願いいたします。

発生している問題・エラーメッセージ

 File "<ipython-input-41-5271bf27d6da>", line 23
    df_2 = df_1.drop([i:])
                                        ^
SyntaxError: invalid syntax

該当のソースコード

for i in range(index, len(df_1.index)):    
    df_2 = df_1.drop([i:])

    #保存
    df_2.to_csv('hoge_{}.csv'.format(i))

試したこと

drop内の[]の削除

Answer 1

import pandas as pd

df = pd.DataFrame({'value': range(10)}, index=range(0,20,2))
#    value
#0       0
#2       1
#4       2
#6       3
#8       4
#10      5
#12      6
#14      7
#16      8
#18      9

# index=6 以降を削除
df.drop(df.loc[6:].index)
#   value
#0      0
#2      1
#4      2

# 6行目 以降を削除
df.drop(df.iloc[6:].index)
#    value
#0       0
#2       1
#4       2
#6       3
#8       4
#10      5

# ループで処理
for i in range(10):
    ret = df.drop(df.iloc[i:].index)

Comments

[SnowBerryTart]

お返事遅くなり申し訳ございません。
ご提示いただいたコードでやりたことが実装できました。
もし良かったら、df.iloc[].indexにしないといけない理由を教えて頂けないでしょうか？

[magichan]

DataFrame.drop() にて複数行の削除を行う場合、パラメータとしては削除する行のラベル値のリストを
df_1.drop([3,4,5])
のように渡す必要がありますので、質問のコードのように
df_1.drop([i:])
スライスにて指定することは残念ながらできません。
そこで、まずは DataFrameを実際にスライス結果のIndex値（のリスト）を
df_1[i:].index
のように求めて、この結果を drop() のパラメータとして渡すことで実現しております。

[SnowBerryTart]

一段階踏む必要があったのですね、非常に納得できました。
丁寧な解説ありがとうございます！