使用`os.listdir`忽略特定文件

Question

如何更改此python代码以忽略Mac OS的.DS_Store文件？

class MySentences(object):
    def __init__(self, dirname):
        self.dirname = dirname

    def __iter__(self):
        for fname in os.listdir(self.dirname):
            for line in open(os.path.join(self.dirname, fname)):
                yield line.split()

编辑：这个问题看起来很像这个问题（ 如何使用os.listdir（）忽略隐藏的文件？ ），但我不知道如何将解决方案实现到上面的类中。 我尝试了这个：

class MySentences(object):
    def __init__(self, dirname):
        self.dirname = dirname

    def __iter__(self):
        for fname in os.listdir(self.dirname) if not fname.startswith('.'):
            for line in open(os.path.join(self.dirname, fname)):
                yield line.split()

但这没有用。

Answer 1

您可以在此处使用if语句，方法是检查文件名不等于（ != ）与“ DS_Store”：

class MySentences(object):
    def __init__(self, dirname):
        self.dirname = dirname

    def __iter__(self):
        for fname in os.listdir(self.dirname):
            if fname != '.DS_Store':
                for line in open(os.path.join(self.dirname, fname)):
                    yield line.split()

如果文件名确实等于该字符串，那么将直接传递该循环，而不会触及嵌套的for语句。