python实现像素值的转换，以下代码有什么错误.

python实现像素值的转换，以下代码有什么错误

import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import numpy as np
import os
from os.path import split,splitext
from PIL import Image

filenames = os.listdir('label_try')
file_path = 'label_try/'
output_dir = 'npy_try'
for i,file_name in enumerate(filenames):
    img = Image.open(file_path + file_name)
    img_arr = np.array(img)
    for index, element in np.ndenumerate(img_arr):
        if element != 0 :
            element = 1
        else:
            pass
    fpath = os.path.join(output_dir,splitext(file_name)[0]+'.npy')
    np.save(fpath, img_arr)

要把一张单通道的图片像素值改为只有0和1

黑色部分为0，白色部分为1

要看你原来的图片，以及你想怎么转
你目前是除非纯白，否则都是黑
if element != 0 :
element = 1
如果你要看到比较有意义的效果，可以计算灰度>50的设置为1，否则设置为0