如何在Python numpy中修改数组元素而不影响复制后的数组？

问题代码 #

import numpy as np  # 补充原代码缺失的导入

R = np.ones(16) * -100
for i in np.arange(10):
    print('Range of 100: ', i , ' Repetitions finished')
    R_save = R
    print(R_save)
    R[0] = 3
    print(R)
    print(R_save)
    comparison = R == R_save
    if comparison.all():
        print('finished Range of 100 potences')
        print(R)
        print(R_save)
        break

问题原因 #

核心问题是NumPy数组属于可变对象，直接用R_save = R赋值时，并没有生成新数组——只是让R_save和R指向内存里的同一个数组对象。所以修改R的元素时，R_save因为共享同一块内存，必然同步变化。

解决方法 #

要让R_save成为独立的数组，必须创建原数组的副本，而非引用。常用三种方式：

• 调用数组自带的copy()方法：R_save = R.copy()
• 使用numpy的copy()函数：R_save = np.copy(R)
• 对一维数组用切片操作：R_save = R[:]

修改后的代码 #

import numpy as np

R = np.ones(16) * -100
for i in np.arange(10):
    print('Range of 100: ', i , ' Repetitions finished')
    R_save = R.copy()  # 替换为创建副本的代码
    print(R_save)
    R[0] = 3
    print(R)
    print(R_save)
    comparison = R == R_save
    if comparison.all():
        print('finished Range of 100 potences')
        print(R)
        print(R_save)
        break

逻辑验证 #

1. 第一次循环：R初始为全-100数组，R_save是它的副本（全-100）；修改R[0]为3后，两者内容不同，不会触发break。
2. 第二次循环：此时R第一个元素已是3，R_save复制该数组；再次修改R[0]为3（无实际变化），此时R和R_save完全相等，触发break，符合你的需求。

内容的提问来源于stack exchange，提问作者Alex