python Конвертировать numpy.ndarray из (4942L, 1L) в (4942L,)

Question

У меня есть два объекта, которые должны быть одинаковой формы.Один из них (4942L, 1L), а другой (4942L,) Как преобразовать массив первого типа, представленный ниже, во второй тип?

#This is what I have:
(array([[0],
       [0],
       [0],
       ..., 
       [0],
       [0],
       [0]], dtype=int64),
 #This is what I need
array([0, 1, 0, ..., 1, 1, 1]))

Answer 1

Предполагая, что ваш массив связан с переменной array, вы можете сделать это с помощью reshape:

new_array = np.reshape(array, (len(array)))

Индексирование также работает:

new_array = array[:, 0]