回答編集履歴

  `iloc[:,i]` に指定している i が DataFrame の列数より多いことにより発生しているエラーです。

 

  `range(train.shape[1])` で `train` の列数でループを回していますが、`test_drop`、`train_drop` がそれより少ない列数だとエラーになります。