python只列出当前目录下的特定条件下的文件

方法一

dir = r'C:\Users\...\_Yin'

def file_filter(file_name):
    if file_name.endswith('.nii'):
        return True
    else:
        return False

files = os.listdir(dir)
files = list(filter(file_filter, files))

print('files:',files)

方法二

import os
output = [dI for dI in os.listdir(dir) if dI.endswith('.nii')]
print(output)