np.reshape 中的 -1 参数

本文详细解析了Numpy中reshape函数的使用方法,尤其对于-1参数的解释,帮助理解如何改变数组形状,同时保持元素数量不变。通过实例展示了如何指定未知维度,由Numpy自动计算。

根据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
来源:知乎
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

<think>我们正在讨论NumPy中的reshape函数,特别是第一个参数设置为-1时的含义。根据引用[2]和引用[4]:当我们在reshape函数中使用-1作为参数时,它表示该维度的大小将由数组原来的大小和其他维度的大小自动推断出来,以保证新数组的元素总数与原数组一致。具体来说,reshape函数的参数-1的作用是:让NumPy自动计算该维度的大小。例如:1.如果我们有一个数组z,其形状为(4,4),即16个元素。2.当我们使用z.reshape(-1)时,它会将数组展平为一维数组,形状为(16,)。3.当我们使用z.reshape(-1,1)时,我们想要一个二维数组,其中第二维的大小为1(即一列),那么第一维的大小就会被自动计算为16(因为16/1=16),所以形状为(16,1)。4.当我们使用z.reshape(2,-1)时,我们指定第一维大小为2,那么第二维的大小就会被计算为8(因为16/2=8),所以形状为(2,8)。因此,-1的作用就是“自动计算该维度的大小”,使得数组元素总数不变。注意:在reshape函数中,最多只能有一个维度指定为-1,因为如果有两个维度都未知,就无法唯一确定。下面我们根据引用[4]中的例子来具体说明:例子:>>>importnumpyasnp>>>z=np.array([[1,2,3,4],[5,6,7,8],[9,10,11,12],[13,14,15,16]])>>>z.shape(4,4)#展平为一维数组>>>z.reshape(-1)array([1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16])#转换为列向量(二维数组,每行一个元素)>>>z.reshape(-1,1)array([[1],[2],[3],[4],[5],[6],[7],[8],[9],[10],[11],[12],[13],[14],[15],[16]])#转换为2行,列数自动计算(8列)>>>z.reshape(2,-1)array([[1,2,3,4,5,6,7,8],[9,10,11,12,13,14,15,16]])总结:在NumPy的reshape函数中,使用-1作为参数可以让该维度的大小由其他维度和总元素数自动推断出来,这样我们就不需要显式地计算该维度的大小,同时保证了数组元素总数不变。相关问题:1.在NumPy中,reshape函数最多可以有几个维度设置为-1?2.使用reshape(-1)和flatten()方法有什么区别?3.如果数组元素总数无法根据给定的维度整除(比如数组有10个元素,reshape(3,-1)),会发生什么?</think>### NumPy中`reshape`函数参数`-1`的含义与作用 在NumPy中,`reshape`函数的第一个参数设为`-1`时,其核心含义是**自动计算该维度的大小**,确保新数组的元素总数与原数组保持一致。具体原理如下: #### 1. **自动维度计算** - 当某个维度指定为`-1`时,NumPy会根据数组的**总元素个数**和**其他已知维度**自动计算出该维度的大小。 - 计算公式为: $$ \text{未知维度大小} = \frac{\text{总元素数}}{\prod(\text{已知维度大小})} $$ - 例如: - 数组`z`形状为`(4, 4)`(共16个元素): ```python z.reshape(-1) # 输出形状:(16,) [展平为一维] z.reshape(-1, 1) # 输出形状:(16,1) [自动计算行数为16] z.reshape(2, -1) # 输出形状:(2,8) [自动计算列数为8] ``` #### 2. **核心规则** - **元素总数守恒**:新数组的元素数必须严格等于原数组元素数,否则报错。 - **唯一未知量**:`-1`只能在**一个维度**中使用(否则无法唯一确定维度大小)。 - **维度兼容性**:已知维度的乘积必须能整除总元素数(如10个元素的数组无法`reshape(3,-1)`)。 #### 3. **典型用例** - **展平数组**: `z.reshape(-1)` → 等价于`z.flatten()`,但更高效(不复制数据)[^1]。 - **生成列向量**: `z.reshape(-1, 1)` → 将任意数组转为$N\times1$的列向量。 - **维度降维**: 若原始维度为`(2,3,4)`,`z.reshape(-1,4)` → 输出形状`(6,4)`(合并前两维)。 #### 4. **与显式维度的对比** | 操作 | 显式写法 | 使用`-1`的等价写法 | |------|----------|-------------------| | 展平 | `z.reshape(16)` | `z.reshape(-1)` | | 转列向量 | `z.reshape(16,1)` | `z.reshape(-1,1)` | | 分两行 | `z.reshape(2,8)` | `z.reshape(2,-1)` | > **关键优势**:无需手动计算维度大小,提升代码的**可读性**和**适应性**(尤其当数据规模动态变化时)[^2][^3]。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值