[np.sum(in_img == i) for i in range(256)]

# 获取输入图像的概率密度
def get_pdf(in_img):
    total = in_img.shape[0] * in_img.shape[1]
    return [np.sum(in_img == i)/total for i in range(256)]  # 求概率密度

pr = get_pdf(p)
print(pr)

输出

但是将上述代码改成下面这样，(为了理解)

def pdf(in_img):
    total = in_img.shape[0] * in_img.shape[1]
    for i in range(256):
        a = np.sum(in_img == i)/total
    return a
print(pdf(p))

输出结果却为 0.0

为啥？？？？？？