numpy(Numerical Python)提供了python对多维数组对象的支持:ndarray,具有矢量运算能力,快速、节省空间。numpy支持高级大量的维度数组与矩阵运算,此外也针对数组运算提供大量的数学函数库。
按照常规做法,我们import numpy as np
,因此下文的np都是numpy的别名
创建ndarray
np.array(data)
- data为数组,几维都可以,用[ ]嵌套括起来就可以。
- 返回一个numpy数组
数组的各种属性
x = numpy.array([1,2,3,4,5])
print(x.dtype) # 打印数组元素类型
print(x.ndim) # 打印数组维度
print(x.shape) # 打印数组各个维度的长度。shape为一个元组
np.newaxis
np.newaxis的功能是插入新维度,以一维数组为例:
x = numpy.array ([1,2,3,4,5])
newaxis放在后面
x[:,numpy.newaxis]
得到二维数组
array([[1],
[2],
[3],
[4],
[5]])
newaxis放在前面
x[numpy.newaxis,:]
得到二维数组
array([[1, 2, 3, 4, 5]]