如何从文件夹直接读取多个csv文件呢?

来源:4-5 tf.data读取csv文件并与tf.keras结合使用

qq_白玉_1

2019-06-23

我们这里直接使用
dataset = tf.data.Dataset.list_files(filenames)
这是因为我们已经有了filenames的列表,但是实际情况我们并没有这个filenames,
只有一个文件夹和其中众多csv文件,那我们怎么读取呢,或者说怎么生成这个所谓的filenames呢

写回答

1回答

qq_白玉_1

提问者

2019-06-24

source_dir = "./generate_csv/"
print(os.listdir(source_dir))

def get_filename_by_prefix(source_dir, prefix_name):
    all_files = os.listdir(source_dir)
    result = [os.path.join(source_dir,filename) for filename in all_files if filename.startswith(prefix_name)]    
    return result

os.listdir(source_dir="xxx")列出文件名列表,需要什么文件自己筛选,这里列出一个根据prefix筛选的。

老师后两节课也有例子

2
0

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

查看课程