Present ndarray of interger elements as ndarray of arrays

Question

I\'m having problems with vectorized function application to ndarrays. What is a good and working way to do this?

Input:

y_train

array([0, 0, 2, 1, 2, 0,