python某些目录中文本文件的数量

Question

我正在尝试获取某个目录中的文件数，但是我希望它仅计算文本文件，因为我在帐户和DS中有另一个目录。 存储文件。 我应该修改什么才能只获得文本文件的数量？

list = os.listdir("data/accounts/")
number_files = len(list)
print(number_files)

Answer 1

从计算Python中具有特定扩展名的文件数开始 ，

解决方案1。

fileCounter = 0
for root, dirs, files in os.walk("data/accounts/"):
    for file in files:    
        if file.endswith('.txt'):
            fileCounter += 1

解决方案2。

fileCounter = len(glob.glob1("data/accounts/","*.txt"))

在这里阅读有关glob的信息

Answer 2

假设您的文本文件以“ .txt”结尾，则可以使用以下内容：

files = [x for x in os.listdir("data/accounts/") if (os.isfile(x) and x.endswith('.txt'))]
number_files = len(list)
print(number_files)

使用os.isfile()忽略目录，使用string.endswith()确定文本文件大小。

Answer 3

可能适合您的替代模块是glob 。

它将允许您使用通配符，以便仅捕获您感兴趣的文件。

from glob import glob
filenames = glob("data/accounts/*.txt")
number_of_files = len(filenames)
print(number_of_files)

Answer 4

number_of_files = sum(f.endswith('.txt') for f in os.listdir("data/accounts/"))

str.endswith()返回True或False
True和False数值分别为1和0。
sum()将使用生成器表达式。