I am trying to delete different parts of a two dimensional array "X" along axis=0 to achieve "X_train". I only know how to delete the different parts one