Numpy常用操作大全

    技术2023-05-26  37

    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')
    Processed: 0.013, SQL: 8