读取文件夹下的地址和文件名

def file_name(file_dir):
    L = []
    for root, dirs, files in os.walk(file_dir):
        for file in files:
            if os.path.splitext(file)[1] == '.csv':
                L.append(os.path.join(root , file))
        return L

file_list = file_name('D:/')
print(np.array(file_list).shape)

读取某个文件夹下的文件列表