如何只获取目录中的文件？

Question

我有这个代码:

allFiles = os.listdir(myPath)
for module in allFiles:
    if 'Module' in module: #if the word module is in the filename
        dirToScreens = os.path.join(myPath, module)    
        allSreens = os.listdir(dirToScreens)

现在,一切正常,我只需要更改线路

allSreens = os.listdir(dirToScreens)

获取只是文件的列表,而不是文件夹.因此,当我使用

allScreens  [ f for f in os.listdir(dirToScreens) if os.isfile(join(dirToScreens, f)) ]

它说

module object has no attribute isfile

注意:我使用的是Python 2.7

Answer 1

你可以使用os.path.isfile方法:

import os
from os import path
files = [f for f in os.listdir(dirToScreens) if path.isfile(f)]

或者,如果你觉得功能:D

files = filter(path.isfile, os.listdir(dirToScreens))

Answer 2

"如果你需要一个文件列表,所有文件名都有一定的扩展名,前缀或中间的任何常用字符串,请使用glob而不是编写代码来自己扫描目录内容"

import os
import glob

[name for name in glob.glob(os.path.join(path,'*.*')) if os.path.isfile(os.path.join(path,name))]