Python日常使用代码

1、找到当前目录下所有符合要求的文件

import os
#找到所有.raw文件并存储起来
f = open("all_raw.txt", 'w')
for root, dirs, files in os.walk('E:\data'):
	for file in files:
		if file[-3:] == 'raw':
			# os.remove(os.path.join(root,file))
            # print(os.path.join(root,file))
			f.writelines(os.path.join(root,file) + '\n')
f.close()

2、将nii文件转换为mhd文件

import SimpleITK as sitk
import numpy as numpy
import os

# 路径和列表声明
# 将PathDicom文件夹下的dicom文件地址读取到lstFilesDCM中
Path = os.path.abspath(r"D:\O") # 与python文件同一个目录下的文件夹,存储dicom文件
save_path = r'E:\A'
list_nii_files = []
for dirName, subdirList, fileList in os.walk(Path):
    for filename in fileList:
        if ".nii" in filename.lower():  # 判断文件是否为dicom文件
            list_nii_files.append(os.path.join(dirName, filename))  # 加入到列表中


for i in range(len(list_nii_files)):
    # itk_img = sitk.ReadImage(list_nii_files[i])
    # img = sitk.GetArrayFromImage(itk_img)
    # print(img.shape)
    folder_path, file_name = os.path.split(list_nii_files[i])
    print(file_name)
    new_path = os.path.join(save_path, file_name[:-4])
    os.mkdir(new_path)
    sitk.WriteImage(sitk.ReadImage(list_nii_files[i]), os.path.join(new_path, "oct.mhd"))#os.path.join(Path,