numpy一维数组永远为列向量

import numpy as np
a = np.array([1,3,4,5])
print(a.shape)
a = np.transpose(a)
print(a.shape)
print(a)

a = np.ravel(a)
print(a.shape)
print(a)

a = a.reshape((1,4))
print(a.shape)

输出如下

(4,)
(4,)
[1 3 4 5]
(4,)
[1 3 4 5]
(1, 4)

我们会发现,对于一维的数组,numpy的数组形式永远是以列向量的形式存储的,不论是使用transpose转置还是ravel尝试flatten这个数组都没办法让列向量转为行向量,这在一般情况下不会产生问题。但在机器学习训练与predict时,有时会要求最后一维的维度,例如为N,那么列向量显然不满足这个维度要求。为此,需要使用reshape方法,让一维numpy列向量转置。

你可能感兴趣的:(python小tips,numpy,numpy,python,机器学习)