Python tips

dict with enumerate

classes= ['0','1','2','3']

class_to_index= {_class: i for i, _class in enumerate(classes)}