在Python中,squeeze()函数用于移除数组中的一维项。具体来说,它将删除形状为1的维度,即删除维度大小为1的维度。
例如,假设有一个形状为(1,5)的数组。使用squeeze()函数后,它将成为一个形状为(5,5)的数组。
将numpy作为np导入
arr = NP . array(【【1,2,3,4,5】】)
打印(排列形状)#输出(1,5)
arr _ squeezed = NP . squeeze(arr)
print(arr _ squeezed . shape)# output(5,)
Squeeze()函数还可以指定删除特定的维,例如,axis=0表示删除第一维。
将numpy作为np导入
arr = NP . array(【【【1,2,3,4,5】】)
print(arr . shape)# output(1,1,5)
arr _ squeezed = NP . squeeze(arr,axis=0)
print(arr _ squeezed . shape)# output(1,5)
以上内容来自互联网,不代表本站全部观点!欢迎关注我们:zhujipindao。com
评论前必须登录!
注册