python中os.path.isfile函数

os.path.isfile():判断某一对象(需提供绝对路径)是否为文件
os.path.isdir():判断某一对象(需提供绝对路径)是否为目录

在学习word2vec中看到这段程序

if os.path.isfile(save_path):  
    raise RuntimeError('the save path should be a dir')
if not os.path.exists(save_path):
    os.mkdir(save_path)

save_path为输入的保存路径。
意思是输入路径若为文件,提示保存路径应该为目录。输入路径不存在,创建目录。

os.listdir():返回一个列表,其中包含有指定路径下的目录和文件的名称
import os
dirct = 'E:\\study\\文本分类\\程序\\tensorflow-word2vec\\model'
for i in os.listdir(dirct):
    print(i)

python中os.path.isfile函数_第1张图片
如上图,运行结果返回5个文件名
checkpoint
params.pkl
tf_vars.data-00000-of-00001
tf_vars.index
tf_vars.meta

我们只需要在使用 os.path.isfile 和 os.path.isdir()时保证输入路径为绝对路径就可以了(使用os.path.join()函数创建,这个函数我在前面有讲os.path.join)

你可能感兴趣的:(python)