Ynum = np.zeros(Y.shape) for num,char in enumerate(np.unique(Y)): Ynum[Y==char] = num