python中想指定某一路径,然后获取该路径下的文件

当某个文件夹里面的文件比较多，自己需要遍历读取，但是又不想一个个自己输入，可以遍历获取存成一个列表，然后自己加一个循环就可以完成多个文件的读取

import os 
# 这里我想指定某一路径,然后获取该路径下的文件
def file_name(file_dir):   
    """ root： 当前目录路径  
        dirs： 当前路径下所有子目录  
        files：当前路径下所有非目录子文件  
    """
    for root, dirs, files in os.walk(file_dir):    
        return files