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

标签： pythonnumpynumpy-ndarray

Answer 1

假设您的数组绑定到变量array，您可以这样做reshape：

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

索引也有效：

new_array = array[:, 0]