如何将标签添加到numpy.ndarray？

Question

我只是想将标签添加到numpy.ndarray。

numpy.ndarray's shape is (?, 1, 100, 100)

[[[ 1 1 1 ... 1 1 1]
          ...
  [ 1 1 1 ... 1 1 1]]]

，标签为[1,0]或[0,1]

所以我想要这样的形状

[[[ 1 1 1 ... 1 1 1]
          ...
  [ 1 1 1 ... 1 1 1]]] , [1,0]

或

[[[ 1 1 1 ... 1 1 1]
          ...
  [ 1 1 1 ... 1 1 1]]] , [0,1]

如何制作此形状？

我尝试过这种方法，但是不起作用。

data_train = []
for i in range(len(true_data.tolist())):
    true_data[i].append([1,0])
    data_train.append(true_data[i])

Answer 1

我自己找到了答案！

这是我的代码

data_train = []
true_list = true_data.tolist()
for index in range(len(true_list)):
    true_list[index].append([1,0])
    temp = np.asarray(true_list[index])
    data_train.append(temp)