求求大佬帮我看看43行这个代码怎么改，自己找了两个小时在网上没找着这样的例子😭

def judge_color(img, mask, colors):
# TODO: 判断亮灯区域颜色
# 提示: mask内颜色均值可以使用cv2.mean()计算
mean = cv2.mean(img)
# 利用颜色距离判断颜色
Dist = np.empty((0,))
for color in colors:
Dist = np.append(Dist, np.sum(abs(mean-color) ** 2))
if Dist[0] < 20: #红色
color_flag = 0;
return mean, color_flag

救命