[新手上路]批处理新手入门导读[视频教程]批处理基础视频教程[视频教程]VBS基础视频教程[批处理精品]批处理版照片整理器
[批处理精品]纯批处理备份&还原驱动[批处理精品]CMD命令50条不能说的秘密[在线下载]第三方命令行工具[在线帮助]VBScript / JScript 在线参考
返回列表 发帖
netdzb

Rank: 5Rank: 5

帖子
909 
积分
1215 
技术
3  
捐助
0  
注册时间
2018-1-17 
1楼 跳转到 » 倒序看帖
打印
tT
发表于 2020-5-26 16:49 | 只看该作者

[问题求助] python遍历文件遇到的问题

def getFiles(dir, suffix): # 查找根目录，文件后缀
    res = []
    for root, directory, files in os.walk(dir):  # =>当前根,根下目录,目录下的文件
            for filename in files:
                    name, suf = os.path.splitext(filename) # =>文件名,文件后缀
                    if suf == suffix:
                            res.append(os.path.join(root, filename)) # =>吧一串字符串组合成路径
    return res


for file_path in getFiles("./", '.png'):  # =>查找以.png结尾的文件
#        img1=Image.open(file_path)
        print(file_path)

当前路径有很多png文件，同时有个子目录101-1，里面也有很多png文件。
我的这段代码会把当前目录下的png和子文件夹的101-1全部列出来的。
如何修改代码，我只要列出当前路径下的png文件，谢谢！
收藏 分享

ivor

Rank: 6Rank: 6

帖子
979 
积分
3381 
技术
172  
捐助
40  
注册时间
2012-1-7 
2楼
发表于 2020-5-26 20:08 | 只看该作者
os.listdir(path='.') 替换 os.walk()
https://docs.python.org/3.8/library/os.html#os.listdir
#&cls&@powershell "Invoke-Expression ([Io.File]::ReadAllText('%~0',[Text.Encoding]::UTF8))" &pause&exit

TOP

Gin_Q

Rank: 4

帖子
264 
积分
330 
技术
13  
捐助
0  
注册时间
2019-9-26 
3楼
发表于 2020-5-26 22:22 | 只看该作者 
  1. directory.clear()
复制代码

TOP

Gin_Q

Rank: 4

帖子
264 
积分
330 
技术
13  
捐助
0  
注册时间
2019-9-26 
4楼
发表于 2020-5-26 22:24 | 只看该作者
本帖最后由 Gin_Q 于 2020-5-26 23:16 编辑

我正好也需要，拿去自己改改！
  1. #!/usr/bin/env python3

  2. #coding=utf-8

  3. 

  4. import os

  5. import time

  6. import shutil

  7. 

  8. 

  9. #禁止访问目录列表

  10. stop = ['837761056','1289832516','1551961057']

  11. #指定复制文件列表

  12. filetype = ['.mp4','.avi']

  13. #复制或移动目标目录

  14. dstdirs = r'd:\\'

  15. 

  16. for root,dirname,filename in os.walk(os.getcwd()):

  17.     print('当前目录为：',root)

  18.     #print('当前目录文件夹有：',dirname)

  19.     #print('当前文件有：',filename)

  20.     #获取当前目录文件不递归

  21.     #dirname.clear()

  22.     #排除指定目录

  23.     for i in stop:  

  24.         if i in dirname:

  25.             dirname.remove(i)

  26.     

  27.     for i in (os.path.join(root,name) for name in filename):

  28.         #print('完整文件路径: ',i,(os.path.getctime(i)))

  29.         for ft in filetype:

  30.             if ft in i:

  31.                 print('正在复制:\n',i,'--->>>',dstdirs)

  32.                 #复制

  33.                 shutil.copy(i,dstdirs)

  34.                 #print('正在移动:\n',i,'--->>>',dstdirs)

  35.                 #移动

  36.                 #shutil.move(i,dstdirs)

  37.     

  38. print('Done!')

  39. os.popen('pause')
复制代码

TOP

返回列表