问答详情
源自:1-4 高效读取30万新闻数据

有没有大神考虑一下放在一起应该怎么写呢,虽然我知道这样更好啊。。但是有没有大神会的。。自己写不明白。。

##貌似是判断文件那有问题?可是不知道怎么写啊。。。有没有大神帮忙看看
class loadFiles(object):
    def __init__(self,par_path):  # par_path 文件夹的路径
        self.par_path = par_path
    def __iter__(self):
        for file in os.listdir(self.par_path):
            print('1. ',file)
            file_abspath = os.path.join(self.par_path,file)
            print('2. ',file_abspath)
            if os.path.isdir(file_abspath):  #如果还是一个文件夹
                loadFiles(file_abspath)
            elif os.path.isfile(file_abspath):  # 如果还是一个文件夹
                for f in os.listdir(file_abspath):  # secondary directory
                    catg = file_abspath.split(os.sep)[-2]
                    print(catg, f)
                    yield catg, f




提问者:慕小小小小小鱼 2020-04-16 20:01

个回答