python实现simpleitk非0置为1

import SimpleITK as sitk
import numpy as np
 
# 加载图像
image = sitk.ReadImage('input.nii.gz')
 
# 将SimpleITK图像转换为NumPy数组
image_array = sitk.GetArrayFromImage(image)
 
# 将非0值置为1
image_array[image_array != 0] = 1
 
# 将NumPy数组转换回SimpleITK图像
image_binary = sitk.GetImageFromArray(image_array)
 
# 保存处理后的图像
sitk.WriteImage(image_binary, 'output.nii.gz')