Numpy中reshape函数用于改变数组形状而不改变数据,新形状元素总数需匹配原数组,如一维12个元素可变为(3,4)或(2,2,3),但不能为(3,5);order参数控制读取顺序,默认’C’行优先;reshape通常返回视图以节省内存,当数据不连续或需重排时返回副本,可通过arr.base判断是否为视图,必要时可用.copy()强制复制。

Numpy中改变数组形状的核心方法就是
reshape
函数。它能让你在不改变数组数据的情况下,以新的维度组织这些数据,就像把一堆积木重新排列成不同的形状,但积木的总数和单个积木本身都没变。
numpy.reshape(a, newshape, order='C')
是这个操作的入口。你传入原始数组
a
,然后指定你想要的新形状
newshape
,它通常是一个表示维度的元组。比如,你有一个包含12个元素的一维数组,你可以把它重塑成
(3, 4)
的二维数组,或者
(2, 2, 3)
的三维数组。
这里有个关键点,新形状的元素总数必须与原始数组的元素总数一致。如果你的原始数组有12个元素,你不能把它重塑成
(3, 5)
,因为
3 * 5 = 15
,这明显不匹配。这是个很常见的错误,新手很容易踩到。
order
参数也值得提一下,它决定了数据在内存中是如何被读取和写入的。
'C'
代表C语言风格的行优先(row-major),也就是最后那个维度变化最快;
'F'
代表Fortran风格的列优先(column-major),第一个维度变化最快。大多数时候我们用默认的
'C'
就够了,但如果你在处理一些科学计算库或者与其他语言接口时,这个参数就显得很重要了。
立即学习“Python免费学习笔记(深入)”;
import numpy as np# 示例1:一维到二维arr1d = np.arange(12)print("原始一维数组:", arr1d)# [ 0 1 2 3 4 5 6 7 8 9 10 11]arr2d = arr1d.reshape((3, 4))print("n重塑为(3, 4)的二维数组:n", arr2d)# [[ 0 1 2 3]# [ 4 5 6 7]# [ 8 9 10 11]]# 示例2:使用-1自动推断arr_unknown_dim = np.arange(15)arr_reshaped_auto = arr_unknown_dim.reshape((3, -1)) # -1 会自动计算为5print("n使用-1自动推断的数组形状:n", arr_reshaped_auto)# [[ 0 1 2 3 4]# [ 5 6 7 8 9]# [10 11 12 13 14]]# 示例3:三维重塑arr_original = np.arange(24).reshape((2, 3, 4))print("n原始三维数组:n", arr_original)# [[[ 0 1 2 3]# [ 4 5 6 7]# [ 8 9 10 11]]## [[12 13 14 15]# [16 17 18 19]# [20 21 22 23]]]arr_new_shape = arr_original.reshape((4, 6))print("n重塑为(4, 6)的二维数组:n", arr_new_shape)# [[ 0 1 2 3 4 5]# [ 6 7 8 9 10 11]# [12 13 14 15 16 17]# [18 19 20 21 22 23]]
Numpy reshape操作会创建新的数组副本还是视图?
这是一个很常见的问题,也挺重要的,因为它直接关系到内存使用和数据修改的副作用。通常情况下,
reshape
会尽量返回一个视图(view),这意味着新的数组对象只是指向了原始数组的相同数据缓冲区。如果原始数组的数据在内存中是连续的,并且新的形状能够以相同的数据布局来解释,Numpy就会很聪明地给你一个视图。这样做的好处是效率高,不占用额外的内存。
但是,也有例外。如果原始数组的数据在内存中不是连续的(比如你对一个数组进行了转置
transpose
操作,或者切片操作导致数据不连续),或者新的形状需要对数据进行重新排列才能满足(例如,你从一个Fortran-order的数组重塑成C-order的数组),那么
reshape
就不得不创建一个副本(copy)。当创建副本时,内存中会有一份新的数据,对新数组的修改不会影响原始数组。
要判断一个
reshape
操作是返回视图还是副本,你可以使用
arr.base is None
或者
arr.base is original_array
来检查。如果
arr.base
不是
None
,并且指向原始数组,那么它就是视图。
import numpy as np# 示例1:通常是视图original_arr = np.arange(12)reshaped_view = original_arr.reshape((3, 4))print("原始数组:", original_arr)print("重塑后的视图:n", reshaped_view)print("reshaped_view是original_arr的视图吗?", reshaped_view.base is original_arr) # True# 修改视图会影响原始数组reshaped_view[0, 0] = 99print("修改视图后,原始数组:n", original_arr) # [99 1 2 3 4 5 6 7 8 9 10 11]# 示例2:何时会创建副本 (例如,需要改变内存布局)# 假设我们有一个非C-contiguous的数组arr_f_order = np.arange(12).reshape((3, 4), order='F')print("nF-order数组:n", arr_f_order)# 重塑成C-order的形状,从F-order到C-order的reshape,如果形状变化,通常会触发copyreshaped_c_order = arr_f_order.reshape((4, 3), order='C')print("reshaped_c_order是arr_f_order的视图吗?", reshaped_c_order.base is arr_f_order) # False# 稳妥起见,如果你想强制创建一个副本,可以使用 .copy()original_arr_for_copy = np.arange(12)reshaped_copy = original_arr_for_copy.reshape((4, 3)).copy()print("reshaped_copy是original_arr_for_copy的视图吗?", reshaped_copy.base is original_arr_for_copy) # False
我个人在实践中,如果我不确定是视图还是副本,或者我明确不希望修改原始数据,我都会习惯性地在
reshape
之后再加一个
.copy()
。这样虽然可能会多一点点内存开销,但能有效
以上就是python numpy如何改变数组的形状_numpy reshape函数改变数组形状的方法的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1372152.html
微信扫一扫
支付宝扫一扫