Split on test and train sets

Всем добра!
Есть массив данных AxBxC и массив меток 1xC. Как в питоне образовать train и test наборы? Я пробовал с помощью cross_validation.train_test_split из sklearn. Да вот незадача, функция, как я понял, для тех случаев, когда вектор признаков для одной метки- это вектор, а не матрица (АхВ), как у меня. Мне б что нибудь, что просто сгенерирует индексы (как в R и Matlab), а не сами массивы дробить будет.