Python 3.x 将1d numpy ndarray转换为形状（n，2）的一个热编码ndarray

我有一个1d numpy阵列，形状

（1,2）

我想把它转换成一个带有形状

（n，2）

的数组，这样它看起来像这样

nd_new = [
            [1,0]
            [1,0]
            [1,0]
            ...
            ...
            [1,0]
         ]

有

n

行

[1,0]

您可以按如下方式使用：

>>> np.tile(nd, (n, 1))

---

我会回答我自己的问题。使用keras提供的one hot编码工具

from keras.utils import to_categorical
y_nd_ones = [1] * n
y_nd = to_categorical(y_nd_ones)

from keras.utils import to_categorical
y_nd_ones = [1] * n
y_nd = to_categorical(y_nd_ones)