I have some data:
data = pd.DataFrame(data=[(1, 1), (2, 1), (2, 5), (0, 2), (1, 4), (0, 1)])
First column should be enum:
clas