python只获取当前目录下的文件夹及文件名

2017-08-16 · 1 min read · python

list = os.listdir(rootdir)#列出目录下的所有文件和目录

1for line in list:
2    filepath = os.path.join(rootdir,line)
3    if os.path.isdir(filepath):#如果filepath是目录
4        print "dir:" + filepath
5    else:
6        print "file:" + filepath

如果需要遍历文件夹下的所以文件，可以使用os.walk()方法。

1os.walk()返回一个三元素的tuple：当前路径、子文件夹名称、文件列表。
2import os
3for root, dirs, files in os.walk(path):
4    for filename in files:
5        print filename
6    for dirname in dirs:
7        print dirname

举个列处当前目录所有文件夹的例子：

1from os import listdir
2from os.path import isfile, join
3import os
4
5dir =os.listdir()
6for line in dir:
7  if os.path.isdir(line):
8    print (line)

参考资料