Попытка добавить столбец в массиве numpy, получая ValueError

Question

Я пытаюсь добавить столбец для следующих массивов

train = np.append(train_data, train_labels, axis=1)

(60000, 784)
(60000,)

И я получаю ошибку

ValueError: all the input arrays must have same number of dimensions

Я не могу понять, в чем проблема ... Мне нужен вывод

(60000, 785)

Answer 1

Вы должны запустить

train = np.append(train_data, train_labels.reshape(-1, 1), axis=1)

Это решит вашу проблему.Ура

Answer 2

Я думаю, что вы должны использовать axis = 0.

>>> np.append((1,2),(2,),axis=0)
array([1, 2, 2])

Если вы опубликовали те кортежи, которые имеют форму массива, то вы можете использовать

train = np.append(train_data, train_labels[:,None], axis=1)

Массивы должны иметьтакое же количество размеров (2 в данном случае).Использование None в индексировании для несуществующего измерения добавляет одноэлементное измерение в массив.

>>> train_labels[:,None].shape
(60000, 1)