[Tips] numpy 收缩维度

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

numpy.squeeze(a,axis = None)

  

 

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

posted @ 2020-03-16 14:12  虚无真仙  阅读(840)  评论(0编辑  收藏  举报