怎么对数据集进行一个裁剪

Fashion-MNIST数据集的训练集和测试集分别有6万和1万张图片10分类的，我怎么才能从每一类中拿出少量数据来训练（比如每一类我拿200张图片）