How to split datasets into specific numbers?

Question

I am using the keras MNIST dataset that comes with a training set of 60k images and a test set of 10k images. In my homework, I am being prompted to split the training set f