根据一维数组的索引和值构造二维numpy数组

根据一维数组的索引和值构造二维numpy数组

也许：

>>> 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.]])