How to write a program to find number of local variables in a function in Python

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

 

Submit Your Programming Assignment Details