Numpy常用操作大全

1. 创建数组

np.array([1,2,3,4], dtype=np.float32)
np.array([[1,2,3,4],[5,6,7,8]])
np.zeros((2,3)) # 生成零元素数组
np.full((2,2),7) # 生成任意元素数组
np.random.rand(2,3) # 生成随机数组
np.eye(3)  # 生成对角矩阵
np.arange(5) #生成从1开始等距数组
np.arange(2,10,2) #生成从2开始到10间距为2的数组
np.linspace(0,1,5) #从0到1间生成大小为5的等距数组

2. 数组操作

a.ndim #数据维度
a.shape # 查看数据各位大小
np.reshape（a, (2,2)） #数据变为2*2的形式
b = a.flatten()  #把数据压缩到1维
np.transpose(a)  #矩阵转置
np.expand_dims(a,axis=0)  #维度扩展
np.squeeze(a, axis=0)  #维度压缩，适合有一维大小为1的矩阵
a[1:5:2] #一维矩阵分割  [start:end:step-size]
a[0,0] #二维位矩阵分割
a[:,-1] #取最后一列
a[:,-1:-3:-1] #取最后一列和倒数第三列，反序列出

3. 矩阵组合

a = np.arange(0,5)
b = np.arange(5,10)
np.vstack((a,b)) #纵向组合
np.hstack((a,b)) #横向组合
np.concatenate(a,b,axis=0) #通过坐标设定组合方向
np.append(a,[[5,6]], axis=0) #尾部追加

4. 取大取小

np.min(a,axis=0) #取小
np.max(a,axis=0) #取大
np.argmin(a,axis=0) #取小坐标
np.argmax(a,axis=1) #取大坐标

5. 排序

np.sort(a, kind='quicksort')