【无标题】随笔2333333

最新推荐文章于 2024-10-08 12:37:10 发布

灵境~

最新推荐文章于 2024-10-08 12:37:10 发布

阅读量95

点赞数

文章标签： Python 遍历目录文件路径写入文件标签

本文链接：https://blog.csdn.net/noelaa/article/details/122329309

版权


```python

import os
import sys
def listfiles(rootDir, txtfile, label=0):
     ftxtfile = open(txtfile, 'w')
     list_dirs = os.walk(rootDir)
     count = 0
     dircount = 0
     for root,dirs,files in list_dirs:
        for d in dirs:
           print(os.path.join(root, d))
           dircount += 1
        for f in files:
            print(os.path.join(root, f))
            ftxtfile.write(os.path.join(root, f)+ ' ' + str(label) + '\n')
            count += 1
            label += 1
     print(rootDir + ' has ' + str(count) + ' files')

listfiles('/home/hhl/hello', '/home/hhl/1')

txt hom/hhl/hello/w3/57.jpg 0
/home/hhl/hello/w3/58.jpg 1
/home/hhl/hello/w3/56.jpg 2
/home/hhl/hello/w1/3.jpg 3
/home/hhl/hello/w1/2.jpg 4
/home/hhl/hello/w1/1.jpg 5
/home/hhl/hello/w2/73.jpg 6
/home/hhl/hello/w2/71.jpg 7
/home/hhl/hello/w2/72.jpg 8