python将numpy.ndarray从(4942L,1L)转换为(4942L,)

时间:2019-03-22 04:00:28

标签: python numpy numpy-ndarray

我有两个需要相同形状的对象。其中之一是(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]))

1 个答案:

答案 0 :(得分:1)

假设您的数组绑定到变量array,则可以使用reshape进行此操作:

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

建立索引也可以:

new_array = array[:, 0]