[Tips] numpy 收缩维度

Posted 2021-03-05 immortalblog

不会改变内部数据的情况下，收缩维度大小为1的维度，使用squeeze函数。

numpy.squeeze(a,axis = None)

　　

 

a表示输入的数组；axis用于指定需要删除的维度，但是指定的维度必须为单维度，否则将会报错；axis的取值可为None 或 int 或 tuple of ints, 可选。若axis为空，则删除所有单维度的条目；
返回值：数组