Python怎么改变NumPy数组的形状（reshape）_NumPy数组维度重塑方法-Python教程-PHP中文网

NumPy数组重塑主要通过.reshape()方法或修改.shape属性实现，前者返回新视图而不改变原数组，后者原地修改形状但可能影响数据独立性；两种方式均要求元素总数不变，否则报错。使用-1可让NumPy自动推断维度大小，但仅能使用一次且需确保整除。reshape通常返回共享内存的视图，修改视图会影响原数组，若需独立副本应使用.copy()，理解视图与副本对性能和数据完整性至关重要。

python怎么改变numpy数组的形状（reshape）_numpy数组维度重塑方法

在Python中，改变NumPy数组的形状（即维度重塑），主要通过两种方式实现：使用

.reshape()

登录后复制

方法，或者直接修改数组的

.shape

登录后复制

属性。这两种方法都能有效地调整数组的布局，但它们在行为和对原始数据的影响上有所不同，理解这些差异对于高效且无误地处理数据至关重要。

NumPy数组的维度重塑，在我看来，是数据处理中一个极其基础但又充满技巧的操作。它允许我们以不同的视角去审视和操作同一份数据，这在机器学习、图像处理等领域简直是家常便饭。比如，你可能从文件中读入一串连续的像素值，但为了用卷积神经网络处理，你必须把它重塑成

（高度，宽度，通道数）

登录后复制

这样的三维结构。

NumPy数组维度重塑的核心操作

我们来深入看看这两种核心操作。

首先是

numpy.reshape()

登录后复制

方法。这是我个人最常用，也推荐新手优先掌握的方式。它的优点在于明确性——它会返回一个具有新形状的数组视图（view），这意味着它通常不会复制底层数据。只有当新形状与原始数组的内存布局不兼容时，它才可能被迫创建一个副本。

立即学习“Python免费学习笔记（深入）”；

import numpy as np

# 原始数组
arr = np.arange(12) # 0, 1, ..., 11
print(f"原始数组：\n{arr}\n形状：{arr.shape}")

# 使用 .reshape() 将一维数组重塑为 3行4列的二维数组
reshaped_arr_1 = arr.reshape((3, 4))
print(f"\n重塑为 (3, 4) 的数组：\n{reshaped_arr_1}\n形状：{reshaped_arr_1.shape}")

# 也可以重塑为 2行2列3深度的三维数组
reshaped_arr_2 = arr.reshape((2, 2, 3))
print(f"\n重塑为 (2, 2, 3) 的数组：\n{reshaped_arr_2}\n形状：{reshaped_arr_2.shape}")

登录后复制

这里有一个关键点，无论是重塑成

（3, 4）

登录后复制

还是

（2, 2, 3）

登录后复制

，所有维度乘积都必须等于原始数组的元素总数（这里是12）。如果元素总数不匹配，NumPy会毫不留情地抛出

ValueError

登录后复制

。

另一种方式是直接修改数组的

.shape

登录后复制

属性。这种方法非常直接，它会原地改变数组的形状。这意味着你不需要创建一个新的变量来接收重塑后的数组，原始数组本身就会被修改。

# 再次使用原始数组，或者重新创建一个
arr_inplace = np.arange(12)
print(f"\n修改前的数组：\n{arr_inplace}\n形状：{arr_inplace.shape}")

# 直接修改 .shape 属性
arr_inplace.shape = (4, 3)
print(f"\n原地修改为 (4, 3) 的数组：\n{arr_inplace}\n形状：{arr_inplace.shape}")

# 尝试不匹配的形状会报错
try:
    arr_inplace.shape = (5, 2) # 5 * 2 = 10, 不等于12
except ValueError as e:
    print(f"\n尝试不匹配的形状时报错：{e}")

登录后复制

我个人觉得，直接修改

.shape

登录后复制

虽然看起来更简洁，但潜在的风险是它会改变原数组，如果你在代码的其他地方还依赖于原数组的旧形状，这可能会引入难以察觉的bug。所以，除非我明确知道需要原地修改，否则我更倾向于使用

.reshape()

登录后复制

来获得一个新视图，这样可以保持原数组的独立性。

NumPy数组重塑时，如何确保数据完整性与性能？

在NumPy数组重塑的过程中，数据完整性通常不是问题，因为

reshape

登录后复制

操作本身并不会改变数组中的元素值，它只是改变了我们“看”这些元素的组织方式。关键在于，重塑后的数组必须包含与原始数组相同数量的元素。如果这个条件不满足，NumPy会直接报错，从某种意义上说，这本身就是一种数据完整性的保护机制。

至于性能，这主要涉及到NumPy的“视图（View）”和“副本（Copy）”机制。大多数情况下，

reshape

登录后复制

操作会返回一个视图。这意味着新数组并没有复制原始数据，而是与原始数组共享同一块内存。这种操作是极其高效的，因为它避免了昂贵的内存分配和数据复制。

import numpy as np

original_arr = np.arange(10)
reshaped_view = original_arr.reshape((2, 5))

print(f"原始数组：{original_arr}")
print(f"重塑后的视图：{reshaped_view}")

# 修改视图中的一个元素
reshaped_view[0, 0] = 99

print(f"修改视图后，原始数组：{original_arr}") # 原始数组也变了

登录后复制

你看，当我们修改了

reshaped_view

登录后复制

中的元素时，

original_arr

登录后复制

也随之改变了。这正是视图的特性，也是其高效的秘密。在我的实践中，我经常利用这种特性来避免不必要的数据复制，尤其是在处理大型数据集时，这能显著提升程序的运行速度。

然而，也有一些情况，

reshape

登录后复制

可能会被迫创建一个副本。这通常发生在原始数组的内存布局不是C-contiguous（行主序）或F-contiguous（列主序）时，或者新的形状要求一个完全不同的内存布局。但对于我们日常使用的

np.arange

登录后复制

、

np.zeros

登录后复制

等创建的数组，默认都是C-contiguous，所以

reshape

登录后复制

通常能返回视图。如果你想强制创建一个副本，可以使用

.copy()

登录后复制

方法，例如

arr.reshape(...).copy()

登录后复制

。这虽然会带来性能开销，但能保证新数组与原数组完全独立。

阿里云-虚拟数字人

阿里云-虚拟数字人是什么？ ...

查看详情

NumPy reshape中的-1参数：维度自动推断的秘密与陷阱

reshape

登录后复制

方法中有一个非常巧妙的特性，那就是可以使用

-1

登录后复制

作为维度参数。这个

-1

登录后复制

的含义是“让NumPy自动计算这个维度的大小”。这在我看来，简直是程序员的福音，它省去了我们手动计算某个维度大小的麻烦，尤其是在处理一些动态数据时，这个特性显得尤为强大。

import numpy as np

data = np.arange(20) # 0到19共20个元素

# 我想把它重塑成4行，但不知道每行多少列
reshaped_auto_cols = data.reshape((4, -1))
print(f"重塑为 (4, -1) 的数组：\n{reshaped_auto_cols}\n形状：{reshaped_auto_cols.shape}")

# 我想把它重塑成5列，但不知道多少行
reshaped_auto_rows = data.reshape((-1, 5))
print(f"\n重塑为 (-1, 5) 的数组：\n{reshaped_auto_rows}\n形状：{reshaped_auto_rows.shape}")

# 甚至可以用于多维重塑
data_3d = np.arange(60) # 60个元素
reshaped_auto_3d = data_3d.reshape((5, -1, 3)) # 5个“块”，每个块3列，中间的行数自动计算
print(f"\n重塑为 (5, -1, 3) 的数组：\n{reshaped_auto_3d}\n形状：{reshaped_auto_3d.shape}")

登录后复制

你看，NumPy非常智能地根据总元素数和已知维度自动推断出了

-1

登录后复制

代表的维度。这确实大大简化了代码，减少了出错的可能。

然而，这个便利的特性也伴随着一些陷阱。最重要的一点是：在一个

reshape

登录后复制

调用中，你只能使用一个
-1
登录后复制
。如果你尝试使用两个或更多的

-1

登录后复制

，NumPy会因为无法确定唯一的解决方案而报错。

try:
    data.reshape((-1, -1)) # 尝试使用两个-1
except ValueError as e:
    print(f"\n尝试使用多个-1时报错：{e}")

登录后复制

另一个需要注意的陷阱是，当使用

-1

登录后复制

时，原始数组的元素总数必须能被所有已知维度的乘积整除。如果不能整除，NumPy同样会抛出

ValueError

登录后复制

。这实际上是

-1

登录后复制

参数工作原理的自然推论——它需要一个明确的、可计算的整数值。

try:
    np.arange(10).reshape((3, -1)) # 10不能被3整除
except ValueError as e:
    print(f"\n元素总数不能被已知维度整除时报错：{e}")

登录后复制

所以，尽管

-1

登录后复制

非常方便，但在使用时我总会确保我的数据量是可预测的，并且与我设定的已知维度能够匹配。这是一种对数据负责的态度，能有效避免运行时错误。

深入理解NumPy数组重塑：视图（View）与副本（Copy）的内存奥秘

视图（View）和副本（Copy）是理解NumPy内存管理的关键概念，尤其在数组重塑时，它们的区别直接影响到程序的性能和行为。我发现很多初学者会在这里栽跟头，因为不理解它们，可能会无意中修改了原始数据，或者进行了不必要的昂贵复制。

视图（View）：一个视图是一个新的NumPy数组对象，但它不拥有自己的数据。相反，它引用了原始数组的底层数据缓冲区。这意味着，如果通过视图修改了数据，原始数组的数据也会随之改变，反之亦然。视图的创建非常高效，因为它不需要分配新的内存或复制数据，仅仅是创建了一个新的“头部”信息来解释同一块数据。

import numpy as np

original_data = np.array([1, 2, 3, 4, 5, 6])
view_data = original_data.reshape((2, 3))

print(f"原始数据：{original_data}")
print(f"视图数据：\n{view_data}")

# 检查是否是视图：如果 .base 属性指向原始数组，则为视图
print(f"view_data.base is original_data: {view_data.base is original_data}")

# 修改视图中的数据
view_data[0, 0] = 99
print(f"\n修改视图后：")
print(f"原始数据：{original_data}") # original_data 也被修改了
print(f"视图数据：\n{view_data}")

登录后复制

在上面的例子中，

reshape

登录后复制

操作返回了一个视图。

view_data.base

登录后复制

属性指向了

original_data

登录后复制

，这明确告诉我们

view_data

登录后复制

是

original_data

登录后复制

的一个视图。

副本（Copy）：一个副本则是一个完全独立的NumPy数组对象，它拥有自己独立的底层数据缓冲区。这意味着，修改副本中的数据不会影响原始数组，反之亦然。创建副本涉及到内存分配和数据复制，这通常是一个相对昂贵的操作，尤其是在处理大型数组时。

那么，什么时候会产生副本呢？

显式请求：你可以通过

.copy()

登录后复制

方法强制创建一个副本。

copied_data = original_data.copy()
print(f"\ncopied_data.base is original_data: {copied_data.base is original_data}") # False
copied_data[0] = 100
print(f"修改副本后，原始数据：{original_data}") # 原始数据未受影响
print(f"副本数据：{copied_data}")

登录后复制