Python教程

Python将list矩阵保存为nii文件

本文主要是介绍Python将list矩阵保存为nii文件，对大家解决编程问题具有一定的参考价值，需要的程序猿们随着小编来一起学习吧！ 
import nibabel as nib
import numpy as np

# 首先读取nii文件并完成维度转换和NAN值替换
path='../data/'
fileName='lactate.nii'
data = nib.load(path+fileName).get_fdata()
data = data.transpose(2, 0, 1)
dataTest = np.nan_to_num(data)

# 某任务产生了新的list
dataTestGT=np.mean(dataTest[:,...],axis=0)[np.newaxis,...]

resultImgs=[]
inputImgs=[]

for index in range(dataTest.shape[0]):
    
    im=dataTest[index]
    gt=dataTestGT[0] 
    
    means = prediction.tiledPredict(...)
    
    resultImgs.append(means)
    inputImgs.append(im)
 
# 保存list为nii
resultImgsshift = np.array(resultImgs).transpose(1, 2, 0)
new_header = nib.load(path+fileName).header.copy()
new_data = np.array(resultImgsshift)
new_img = nib.nifti1.Nifti1Image(new_data, None, header=new_header)
nib.save(new_img, 'N2VPrediction.nii')

这篇关于Python将list矩阵保存为nii文件的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！
您可能喜欢
栏目导航