问题

代码中出现了 sqrt(4)，报错信息：

Undefined variable 'sqrt'

这是因为没有定义sqrt函数，引入math库就可以解决。

解决

在代码开头写上

from math import sqrt

再次运行，不再报错。