python 获取指定文件夹下 指定后缀 的文件名集合列表

def fileNames(root, suffix=None):
    names = os.listdir(root)
    result = []
    if suffix:
        for name in names:
            if os.path.splitext(name)[1] == suffix:
                result.append(name)
    else:
        result = names
    return result