Python numpy为数组增加一个新维度--np.newaxis和np.expand_dims

np.newaxis 可以帮助我们为数组增加一个新维度,比如说,将一个向量转换成列矩阵和行矩阵:

首先导入numpy

import numpy as np

后续代码如下:





高维数组增加维度:




增加维度也可以用np.expand_dims




猜你喜欢

转载自blog.csdn.net/lcxxcl_1234/article/details/80726604