python实现将二维矩阵m * n补零为方阵n * n

思路:

  • 先将二维矩阵m * n转化为列表

  • 进行补零操作

  • 最后再次转化为数组

data = data.tolist() ##作为矩阵的data,先将其转化为列表
                ###添加0,使二维温度矩阵变为方阵
                for item1 in data:
                    if(len(item1) != len(data)):
                        if(len(item1) < len(data)):
                            for _ in range(len(data) - len(item1)):
                                item1.append(0)  ##矩阵行大于列,对列进行补零
                        else:
                            data.append([0 for _ in range(len(item1))] for _ in range(len(item1) - len(data)))  ##矩阵列大于行,对行进行补零
                            break
                data1 = np.array(data)  ###作为方阵的data1

你可能感兴趣的:(python学习,神经网络数据准备,python,数据处理,矩阵)