numpy数组的变形

numpy数组的变形

个人学习笔记！

常用的两种变形方法： reshape()函数、newaxis关键字

一、reshape()

使用条件：原始数组的大小（size）必须和变形后的一致。

注意：reshape()方法会用到原始数组的非副本视图

>>> grid = np.arange(1,10).reshape((3,3))
>>> grid
array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])

二、newaxis

newaxis将一个一维数组变成二维的行或列的矩阵

>>> x = np.array([1,2,3])

>>> y=x [ np.newaxis , : ]   #变成二维的行向量

>>> y
array([[1, 2, 3]])

>>> y=x[ : , np.newaxis ]   #变成二维的列向量
>>> y
array([[1],
       [2],
       [3]])