How to set the listdir in for loop?

See more:

Python

import os

subDir = []
def listdirs(path, count, which_dir):
    for roots, dirs, files in os.walk(path):
        for dir in dirs:
            d = os.path.join(roots, dir)
            subDir.append(d)
    b = [s.split() for s in subDir]
    return b[which_dir]

rootdir = '/home/runner/TestP1'
count = 0
directory = listdirs(rootdir, count, 1)
which = len(subDir)

print(f"Total directories: {which}")
print(f"Directory: {directory}")

What I have tried:

In this code, I want to set the directory = listdirs(rootdir, count, 1) in for loop so that I could print all the paths and use them further one by one. But the problem is that I wrote which variable above the listdirs() function like this,

rootdir = '/home/runner/TestP1'
count = 0
which = len(subDir)
print(f"Total directories: {which}")
for x in range(which):
    directory = listdirs(rootdir, count, x)
    print(f"Directory: {directory}")

But it gave me this result

Total directories: 0

It didn't show me the a directory