如何从文件夹直接读取多个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筛选的。
老师后两节课也有例子
20