Splitting dataset into train, test and validation sets to maintain proportions

Question

I am working with the iWildCam dataset, and I am trying to split my dataset so that it maintains the proportions of animals in the train, test and validation sets, while at