python count number of text file in certain directory

Question

I am trying to get the number of the files in a certain directory, but I want it to count only text file because I have another directory in the accounts and DS. Store file. What should I modify to only get the number of text files?

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

Answer 1

From Count number of files with certain extension in Python,

Solution 1.

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

Solution 2.

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

Read about glob here here

Answer 2

Assuming your text files end in ".txt", you could use something like:

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

Using os.isfile() to ignore directories and string.endswith() to determine text file-ness.

Answer 3

An alternative module that may work for you is glob.

It will allow you to use wildcards so you can capture just the files you are interested in.

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() returns True or False
True and False have numeric values of one and zero, respectively.
sum() will consume the generator expression.