Python：创建一个类来读取目录中的文件

Question

我正在尝试自学如何在Python中创建类。我目前能够编写一个类，使我能够使用相同的类来绘制多个形状，这是非常基本的。

我遇到使用类的更高级任务的问题，例如我希望能够创建一个包含特定目录中有多少文件等功能的类。

我想我的问题是如何创建一个读入目录的类并告诉我目录中有多少文件？

谢谢，对不起，如果不清楚

Answer 1

os.listdir()返回目录的内容列表。如果我正确理解了这个问题。

所以如果你想为它创建一个新类，那么一个类将保存该函数的函数包装器。不确定“读入目录”是什么意思

class YourClass:

    def __init__(self):
         #initialize here

    def countFilesInDir(self,directory):
         return len(os.listdir(directory))

或类似的东西

Answer 2

您正在寻找的功能是os.listdir。

给定目录/tmp/foo，其中包含名为bar和baz的文件：

>>> import os
>>> os.listdir('/tmp/foo')
['baz', 'bar']

当您使用os.listdir时，您还需要使用os.path.isfile来测试您正在处理的每个条目是否实际上是一个文件，因为还会返回目录名称。