Python中yield使用方式

最新推荐文章于 2023-12-26 20:00:00 发布

汤姆是猫？

最新推荐文章于 2023-12-26 20:00:00 发布

阅读量171

点赞数

分类专栏： Python脚本

本文链接：https://blog.csdn.net/painprogrammer/article/details/103701080

版权

Python脚本专栏收录该内容

7 篇文章 0 订阅

订阅专栏

利用yield的函数遍历目录,实现按顺序处理文件,我在这里处理的是xml,可以根据需要改变!

def xml1(path):
    for i in range(0, len(os.listdir(path))):
        yield os.path.join(path, str(i) + '.xml')


def xml2(path):
    for i in range(1, len(os.listdir(path))):
        yield os.path.join(path, str(i) + '.xml')


if __name__ == "__main__":
    # 传入xml的路径
    path = ''
    path1 = xml1(path)
    path2 = xml2(path)
    i = 0
    while True:
        if i >= len(os.listdir(path)) - 1:
            break
        print(next(path1), next(path2))
        i += 1