也许:
>>> Y = np.array([2, 0, 1, 1])>>> X = np.zeros((len(Y), 3))>>> X[np.arange(len(Y)), Y] = 1>>> Xarray([[ 0., 0., 1.], [ 1., 0., 0.], [ 0., 1., 0.], [ 0., 1., 0.]])
欢迎分享,转载请注明来源:内存溢出
也许:
>>> Y = np.array([2, 0, 1, 1])>>> X = np.zeros((len(Y), 3))>>> X[np.arange(len(Y)), Y] = 1>>> Xarray([[ 0., 0., 1.], [ 1., 0., 0.], [ 0., 1., 0.], [ 0., 1., 0.]])
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)