python小工具

Posted 2021-09-05 Mario cai

将数组保存为TXT

np.savetxt("./result.txt", img_arr,fmt ='%d')

将数组保存为图片

import numpy as np

import matplotlib.pyplot as plt

x = np.array([[1, 2, 3], [4, 5, 6]], np.int32)

plt.imshow(x)

plt.savefig("array")

nii变RGB（先读取数组再成像再保存）

import nibabel as nib
import numpy as np
import os
import matplotlib.pyplot as plt


file = r'C:\\新建文件夹/mask/1.nii'
#filenames = os.listdir(file)
img = nib.load(file)

nii = img.dataobj

(x, y, z) = nii.shape

for i in range(0, 200):

    #img_path = os.path.join(file, f)
    img_arr = nii[:, i, :]
    plt.imshow(img_arr)
    plt.savefig('%d.png'%(i+1))