给定系统脉冲响应代码报错排查：NameError及变量未定义问题解决

我编写了一段以脉冲向量为输入、返回脉冲响应向量的代码：

import numpy as np
from scipy import signal


y = signal.unit_impulse(8)

def response(y):
      for i in range(8):
      g = np.zeros(len(y))
      g[i] = 3 * y[i] -2 * y[i - 1] + 2 * y[i -2]
return g

print(g)

其中signal.unit_impulse(8)用于生成一个8位向量，首位为1，其余为0。运行代码时出现错误NameError: name 'g' is not defined，函数逻辑内还收到提示“Local variable 'g' might be referenced before assignment”，这两个提示相关，请问该如何修复？

1. 解决变量作用域与初始化问题 #

• 函数内的g必须在循环外初始化：原代码中每次循环都会重新创建全零数组g，若循环未执行（极端情况），g会处于未定义状态，触发"might be referenced before assignment"警告。
• 全局作用域无法直接访问函数内的局部变量g：原代码的print(g)直接调用未在全局定义的变量，必须先调用函数并接收返回值，再打印该返回值。

2. 处理边界索引异常 #

当i=0或i=1时，y[i-1]、y[i-2]会取到负索引，numpy会默认从数组末尾取值，这不符合脉冲响应的计算逻辑，需要给边界情况做特殊处理（索引小于0时取值为0）。

import numpy as np
from scipy import signal

y = signal.unit_impulse(8)

def response(y):
    # 在循环外初始化响应数组g
    g = np.zeros(len(y))
    for i in range(len(y)):
        # 处理边界索引，索引小于0时y的取值为0
        y_prev1 = y[i-1] if i >= 1 else 0
        y_prev2 = y[i-2] if i >= 2 else 0
        g[i] = 3 * y[i] - 2 * y_prev1 + 2 * y_prev2
    return g

# 调用函数并接收返回值，再打印结果
output = response(y)
print(output)

• 将g的初始化移至循环外，确保函数内始终有g的定义，消除未赋值就引用的警告。
• 通过调用函数并赋值给output，再打印该变量，解决全局作用域找不到g的NameError。
• 添加边界索引判断，避免负索引导致的错误计算，贴合脉冲响应的实际逻辑。

内容的提问来源于stack exchange，提问作者In the blind