Python获取文件夹下所有文件类型-用集合删除重复项

2018-03-19  本文已影响18人  莫尛莫

1. 遍历文件目录

import os
for dirpath, dirnames, filenames in os.walk('D:\迅雷下载'):
    print(dirpath)
    print(dirnames)
    print(filenames)

2. 创建集合

set1 = set()    # 新建
set2 = {1,2,3}    # 新建包含值 

set1.add(5)    # 增加一个元素5
set.update([5,6,7,8])    # 增加多个元素

set1.remove(1)    # 删除元素“1”

3. 拆分文件名和后缀

os.path.splitext(file)[0]
os.path.splitext(file)[1]

4.完整代码

import os
set1 = set()

def getFile(sourcePath):
  for root, dirs, files in os.walk(sourcePath):
      for file in files:
          set1.add(os.path.splitext(file)[1])


getFile('D:\迅雷下载')
print(set1)
上一篇下一篇

猜你喜欢

热点阅读