Question

python怎么增加矩阵维度

Answer 1

• numpy.expand_dims(a, axis) 
  Examples

• >>> x = np.array([1,2])


• >>> x.shape


• (2,)


• >>> y = np.expand_dims(x, axis=0)


• >>> y


• array([[1, 2]])


• >>> y.shape


• (1, 2)


• >>> y = np.expand_dims(x, axis=1)  # Equivalent to x[:,newaxis]


• >>> y


• array([[1],


•       [2]])


• >>> y.shape


• (2, 1)