how Splitting Dataset by class?

Question

I want to separate my data set by class. Get a dataset that has 1000 examples of each . My classes are: 0-1-2-3-4-5-6-7-8-9;

import tensorflow 

import numpy