python - 无法将 (N, 1) 数组转换为 (N,)

Question

我有一个 (200, 1) 矩阵，我需要将其转换为 (200,)

y = np.matrix(np.append(np.zeros(200), np.ones(200))).T

y 的形状为 (200,1)

我尝试压平矩阵:

c = y.flatten() 

我得到了 (1,200) 的形状

然后我尝试访问矩阵的第一个元素，

c[0].shape

我仍然保持形状 (1, 200)

如何将原始矩阵转换为(200,)

Answer 1

您可以使用y.A1

>>> y = np.matrix(np.identity(3))
>>> y
matrix([[1., 0., 0.],
        [0., 1., 0.],
        [0., 0., 1.]])
>>> y.A1
array([1., 0., 0., 0., 1., 0., 0., 0., 1.])