@R星校长
第2
关:形状操作
怎样改变ndarray
对象的形状
改变形状
上一关介绍了怎样实例化ndarray
对象,比如想实例化一个3
行4
列的二维数组,并且数组中的值全为0
。就可能会写如下代码:
import numpy as np
a = np.zeros((3, 4))
那如果想把a
变成4
行3
列的二维数组,怎么办呢?比较聪明的同学可能会想到这样的代码:
import numpy as np
a = np.zeros((3, 4))
# 直接修改shape属性
a.shape = [4, 3]
最后你会发现,这样的代码可以完成功能,但是这种直接改属性的方式太粗暴了,不符合良好的编程规范。
更加优雅的解决方式是使用NumPy
为我们提供了一个用来改变ndarray
对象的shape
的函数,叫 reshape
。
NumPy
为了照顾偏向于面向对象或者这偏向于面向过程这两种不同风格的程序员,提供了2种调用**reshape函数的方式(**其实很多函数都提供了两种风格的接口)。
如果你更偏向于面向对象,那么你可以想象成ndarray
对象中提供好了一个叫reshape
成员函数。代码如下:
import numpy as np
a = np.zeros