也许：
>>> 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.]])
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
