python中获取目录下的文件数

Python里面用来遍历目录的os.walk()方法，os.walk会返回一个生成器，所以可以用在for循环。

     os.walk(dir)会返回一个三元组：（当前目录，子目录列表，文件列表）

1.获取目录下的文件名

#---encoding:UTF-8----
#!/usr/bin/python
import sys,os
def fileCountIn(dir):
    return sum([len(files) for root,dirs,files in os.walk(dir)])
#os.walk(dir)会返回一个三元组：（当前目录，子目录列表，文件列表）
#所以len(files)就是获取当前目录下的文件数，然后每个目录下的文件数求和即可
if __name__=='__main__':
    if len(sys.argv)==2:
 dir=sys.argv[1]
 print 'Total files in %s is: %d' %(dir,fileCountIn(dir))
    else:
 print 'usage:filecount.py dirname'

2. 用于查看输入路径中的目录以及目录中的所有文件

#encoding:UTF-8
import  string,os, sys
dir=raw_input('请输入路径名:')
print '----------- no sub dir（不含子目录）'
files =os.listdir(dir)
for f in files:
    print dir + os.sep + f
print '----------- all dir（包含子目录及其下所有文件）'
for root, dirs, files in os.walk(dir):
    for name in files:
        print os.path.join(root, name)