How to make an argument to print the desired pathname?

See more:

Python

import os

subDir = []
def listdirs(rootdir, count):
    def recurse(rootdir):
        nonlocal count
        for file in os.listdir(rootdir):
            d = os.path.join(rootdir, file)
            if os.path.isdir(d):
                recurse(d)
                count += 1
                subDir.append(d)
    recurse(rootdir)
    print(f"Total directories are {count}")

rootdir = ('/home/runner/TestP1')
count = 0
listdirs(rootdir, count)
print(subDir[2])

What I have tried:

In this code, I am taking the rootdir, counting the subdirectories, and printing them by my choice. I used subDir to print the desired subdirectory. But I want to use the subDir as an argument. I tried it but failed.

How can I use subDir as an argument and print the desired result just like it is printing it without argument?