python - 如何计算python目录中具有2种扩展名的文件？

Question

我有一个包含多种类型文件的目录。如何计算具有 2 种扩展名(.txt 和 .csv)的目录中的文件数？在我的搜索中，我发现了如何只用一个特定的扩展名来计数

Answer 1

假设 path 是您文件夹的路径。然后

import os

# get list of files
list_of_files = os.listdir(path)

# txt files
num_txt = len([x for x in list_of_files if x.endswith(".txt")])
# csv files
num_csv = len([x for x in list_of_files if x.endswith(".csv")])