当前位置: 代码迷 >> 综合 >> np.reshape 中的 -1 参数
  详细解决方案

np.reshape 中的 -1 参数

热度:90   发布时间:2023-12-25 19:28:15.0

根据Numpy文档(https://docs.scipy.org/doc/numpy/reference/generated/numpy.reshape.html#numpy-reshape)的解释:
newshape : int or tuple of ints
The new shape should be compatible with the original shape. If an integer, then the result will be a 1-D array of that length. One shape dimension can be -1. In this case, the value is inferred from the length of the array and remaining dimensions.

?? 大意是说,数组新的 shape 属性应该要与原来的配套,如果等于 -1 的话,那么Numpy会根据剩下的维度计算出数组的另外一个 newshape 属性值。

举几个例子或许就清楚了,有一个数组 z 如下:

??而 z.reshape(-1, 1) 是说,我们不知道新z的行数是多少,但是想让z变成只有一列,行数不知的新数组,通过z.reshape(-1,1),Numpy 自动计算出有 16 行,新的数组 shape 属性为 (16, 1),与原来的 (4, 4) 配套。

z.reshape(-1, 2),行数未知,列数等于 2,reshape 后的 shape 等于 (8, 2)

同理,只给定行数,列数未知,也可以设置 newshape 等于 -1,Numpy 也可以自动计算出新数组的列数。

作者:李彬
链接:https://www.zhihu.com/question/52684594/answer/157491724
来源:知乎
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。