python学习笔记——numpy的squeeze方法

squeeze的中文意思是挤压的意思
在numpy中squeeze的用法是什么呢？
首先运行代码

第一次代码实验

cost=[[[[12,24]]]]
#cost=[[12,22],[23,33],[34,44]]
#
print('cost,squeeze之前的样子',cost)
print(type(cost))
#print(len(cost))
#print(cost[0][1])
cost=np.squeeze(cost)
print("________________________")
print("squeeze之后是什么样子：",cost)
print(type(cost))
print("hang:",cost.shape[0])
#print("lie:",cost.shape[1])

运行后的结果，会把所有的中括号都挤压掉，就剩下一个中括号
是一个2维的列向量

第二次代码实验

import numpy as np
cost=12
#
print('cost,squeeze之前的样子',cost)
print(type(cost))
#print(len(cost))
#print(cost[0][1])
cost=np.squeeze(cost)
print("________________________")
print("squeeze之后是什么样子：",cost)
print(type(cost))

这次定义了一个变量cost=12，是int类型的，squeeze之后，cost还是等于12，只是数据类型变为ndarray

第三个实验

import numpy as np
cost=[[12],[23],[34]]
#
print('cost,squeeze之前的样子',cost)
print(type(cost))
#print(len(cost))
#print(cost[0][1])
cost=np.squeeze(cost)
print("________________________")
print("squeeze之后是什么样子：",cost)
print(type(cost))

运行后的结果，把里面的三个中括号都删除了

第四个实验

cost=[[12,22],[23,33],[34,44]]
#
print('cost,squeeze之前的样子',cost)
print(type(cost))
#print(len(cost))
#print(cost[0][1])
cost=np.squeeze(cost)
print("________________________")
print("squeeze之后是什么样子：",cost)
print(type(cost))
print("hang:",cost.shape[0])
print("lie:",cost.shape[1])

变成了一个3行两列的矩阵