【每日一坑 4】 查找文件

南斗

1. #! /usr/bin/env python
   
2. # -*- coding: utf-8 -*-
   
3. import os
   
4. rootdir = 'D:\AutoTest' #定义根目录
   
5. i = 0
   
6. 
   
7. #三个参数：分别返回1.父目录 2.父目录下所有文件夹名字（不含路径） 3.父目录下所有文件名字
   
8. for dirpath, dirnames, filenames in os.walk(rootdir):
   
9.     #print filenames
   
10.     for filename in filenames: #遍历文件名称
    
11.         #print filename
    
12.         if filename.split('.')[1] == 'pyc': #提取文件扩展名并进行比较
    
13.             i = i + 1
    
14. print '后缀是pyc的文件一共有：%d个' % i

复制代码