np.squeeze
:从数组的形状中删除维度为 \(1\) 的维度。
np.squeeze(arr, axis)
参数:
arr
:输入数组axis
:整数或整数元组,用于选择形状中一维维度的子集。
示例:
import numpy as np
x = np.arange(9).reshape(1, 3, 3)
print(x)
y = np.squeeze(x)
print(y)
print(x.shape, y.shape)
[[[0 1 2]
[3 4 5]
[6 7 8]]]
[[0 1 2]
[3 4 5]
[6 7 8]]
(1, 3, 3) (3, 3)
在用 matplotlib
直接画图,会报错,可以利用squeeze()
函数将表示向量的数组转换为秩为1的数组。
plt.plot(x)
plt.show()
ValueError: x and y can be no greater than 2-D, but have shapes (1,) and (1, 3, 3)
plt.plot(y)
plt.show()
标签:维度,plt,print,数组,np,squeeze 来源: https://www.cnblogs.com/keye/p/16619241.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。