numpy学习-np.expand_dims()增加一维

np.expand_dims(narray, axis=):用于扩展数组的形状

初始形状

import numpy as np
a = np.array([[1,1,1],[1,1,1]])
print(a.shape)
>>>(2, 3)
print(a)
>>>[[1 1 1]
 [1 1 1]]

在0轴添加元素

b = np.expand_dims(a,axis=0)
print(b.shape)
>>>(1, 2, 3)
print(b)
>>>[[[1 1 1]
  [1 1 1]]]

在1轴添加

c = np.expand_dims(a,axis=1)
print(c.shape)
>>>(2, 1, 3)
print(c)
>>>[[[1 1 1]]

 [[1 1 1]]]

在2轴添加

d = np.expand_dims(a,axis=2)
print(d.shape)
>>>(2, 3, 1)
print(d)
>>>[[[1]
  [1]
  [1]]

 [[1]
  [1]
  [1]]]