Python中仅更新AverageU时StartingU同步改变的代码问题排查

问题表现 #

基于3×3初始数组StartingU计算size=2、步长stride=1的滑动窗口平均值，结果存入AverageU数组时，代码执行后从未被主动赋值修改的StartingU没有保留初始值，和AverageU同步更新为计算后的结果，不符合预期。

核心错误点 #

• 代码中先调用np.zeros初始化了独立的3×3零矩阵AverageU，但后续执行AverageU = StartingU的直接赋值操作时，没有产生任何数组数据的拷贝，只是让AverageU变量指向了和StartingU完全相同的内存地址
• 这行赋值执行后，最开始初始化的零矩阵已经被废弃，后续所有对AverageU元素的修改，本质都是直接修改StartingU对应内存位置的值，自然会出现两个数组同步变化的现象

修正方法 #

• 不要用直接赋值的方式把原数组绑定给结果数组，结果数组要初始化独立的内存空间：可以直接用np.zeros_like(StartingU, dtype=np.float64)生成和原数组形状一致的独立零矩阵，因为平均值是浮点数，要指定浮点类型避免整数截断误差
• 如果需要基于原数组的值做计算初始化，必须用显式拷贝方法生成独立数组，比如AverageU = StartingU.copy()，拷贝后两个数组内存完全独立，修改结果不会影响原数组

修正后可运行代码 #

import numpy as np

# 初始化原始数组
StartingU = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 滑动窗口参数
window_size = 2
stride = 1
# 初始化独立的结果数组
AverageU = np.zeros_like(StartingU, dtype=np.float64)

# 打印计算前的数组状态
print("计算前StartingU:")
print(StartingU)
print("计算前AverageU:")
print(AverageU)

# 获取数组形状，遍历滑动窗口
height, width = StartingU.shape
for i in range(0, height - window_size + 1, stride):
    for j in range(0, width - window_size + 1, stride):
        # 截取当前窗口区域
        current_window = StartingU[i:i+window_size, j:j+window_size]
        # 计算平均值写入结果数组对应位置
        AverageU[i, j] = current_window.mean()

# 打印计算后的数组状态
print("计算后StartingU（保持初始值）:")
print(StartingU)
print("计算后AverageU:")
print(AverageU)

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