Python program to find number of local variables in a function

Given a Python program, task is to find the number of local variables present in a function.

Examples:

Input : a = 1
        b = 2.1
        str = 'GeeksForGeeks'
    
Output : 3

We can use the co_nlocals() function which returns the number of local variables used by the function to get the desired result.

Code #1:

# Implementation of above approach

# A function containing 3 variables
def fun():
    a = 1
    str = 'GeeksForGeeks'


# Driver program
print(fun.__code__.co_nlocals)

Output:

2

Code #2:

# Python program to find number of
# local variables in a function

# A function containing no variables
def geek():
    pass

# A function containing 3 variables
def fun():
    a, b, c = 1, 2.25, 333
    str = 'GeeksForGeeks'

# Driver program
print(geek.__code__.co_nlocals)
print(fun.__code__.co_nlocals)

Output:

0
4