Question

python获取文件夹中的图片的路径

/home/zyy/汽车/卡槽这个文件夹下有一个图片1.jpg和一个231.txt还有一些其他文件 /home/zyy/汽车/山地这个文件夹下同样有个图片1.jpg和一些其他文件 如何用python获取/home/zyy/汽车这个文件夹以及子文件（卡槽，山地等）的所有图片的路径

Answer 1

Python 3.6.1 (default, Mar 22 2017, 06:17:05) 
[GCC 6.3.0 20170321] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import os
>>> def isimage(fn):
...     return os.path.splitext(fn)[-1] in ('.jpg', '.JPG', '.png', '.PNG')
... 
>>> isimage('abs.jpg')
True
>>> isimage('abc.txt')
False
>>> dirpath = '/home/zyy/汽车/卡槽'
>>> for r, ds, fs in os.walk(dirpath):
...     for fn in fs:
...             if not isimage(fn):
...                     continue
...             fname = os.path.join(r, fn)
...             print(fname)
...

追问

厉害大神 如何把这个fname输出保存在txt文档里呢

追答

#!/usr/bin/env python3
import os


dirpath = '/home/zyy/汽车/卡槽'
targetfile = 'imagepath.txt'


def isimage(fn):
    return os.path.splitext(fn)[-1] in (
        '.jpg', '.JPG', '.png', '.PNG')

        
def main():
    imagelist = []
    for r, ds, fs in os.walk(dirpath):
        for fn in fs:
            if not isimage(fn):
                continue
            fname = os.path.join(r, fn)
            print(fname)
            imagelist.append(fname)
    if not imagelist:
        print('image not found')
        return
    target = os.path.join(dirpath, targetfile)
    with open(target, 'w') as f:
        f.write('\n'.join(imagelist))
    print('the path of images have been wirte to', 
        target)
        
        
if __name__ == '__main__':
    main()

追问

大神 终端里输出是
/home/zyy/汽车/卡槽/1.jpg
/home/zyy/汽车/山地/1.jpg
但是文本里只写了一个
/home/zyy/汽车/卡槽/1.jpg
怎么修改