Split data to 'classes' with pandas or numpy,[x.index.tolist() for x in dfs]