一、numpy.array 增加维度

import numpy as np

a = np.array([1, 2, 3])

print(a)

#输出

array([1, 2, 3])

print(a[None])

#输出

array([[1, 2, 3]])

print(a[:,None])

#输出

array([[1], [2], [3]])

二、复制填充并升维

举例：把a的shape从(10，10)变成(5，10，10)

np.expand_dims(a,0).repeat(1000,axis=0)

⚠️ expand_dims表示增加一个维度，上式增加在a的0维度。

repeat代表重复的次数，axis代表在哪个维度进行重复，注意index从零开始，-1是增加在最后一维。

三、压缩冗余维度

在数组中会存在很多轴只有1维的情况，使用squeeze函数来压缩维度。

import numpy as np

b = np.array([[[[1],[2]],[[3],[4]]]])

b.shape

# 输出

(1L, 2L, 2L, 1L)

print(b)

# 输出

array([[[