numpy array保存为nii格式

将.nii文件读取成numpy array是很常见的，但将分割好的结果numpy array保存为nii格式却也是必须的。
现在介绍两种将numpy array保存为nii格式的方法。

1、SimpleITK

## using simpleITK to load and save data.
import SimpleITK as sitk
itk_img = sitk.ReadImage('./nifti.nii.gz')
img = sitk.GetArrayFromImage(itk_img)
print("img shape:",img.shape)

## save 
out = sitk.GetImageFromArray(img)
sitk.WriteImage(out,'simpleitk_save.nii.gz')

2、Nibabel

import nibabel as nib 
import numpy as np 
 
new_image = nib.Nifti1Image(my_arr, np.eye(4)) 
nib.save(new_image, 'nifti.nii.gz')