CODEWITHSUNDEEP
pythonscope
brownser
brownser

Reputation: 622

How to change a variable inside a function scope after defining the function in Python?

I'm looking for a way to change the variables defined inside a function after defining the function.

For example 

def GetNthPower(x) :
    n = None
    return x**n

my_numbers_list = [11,23,45,56,78,98]

# now if I feel like I need the 4th power of some numbers in the list 

GetNthPower.n = 4

for x in my_numbers_list :
    print GetNthPower(x)

#  If I want 7th power then
GetNthPower.n = 7

This obviously will not work, is there any way to do this?

N.B: I know we can achieve this by setting 'n' as an argument of the function, but I want to do it this way for a particular reason. I want my function to have only one argument (for using the function in multiprocessing.Pool.map()).

Upvotes: 1

Views: 104

Answers (3)

C.Nivs
C.Nivs

Reputation: 13106

Alternatively, you could use functools.partial to bind a keyword argument to a function

from functools import partial

def get_nth_power(x, n):
    return x ** n

get_third = partial(get_nth_power, n=3)

get_third(4)
64

x = 4

# in a loop
for pow in [2, 4, 6, 8]:
    f = partial(get_nth_power, n=pow)
    f(x)

Upvotes: 2

chepner
chepner

Reputation: 531305

Don't use one function; create a function that makes your function, using a closure.

def nth_power_maker(n):
    def _(x):
        return x ** n
    return _

my_numbers_list = [11,23,45,56,78,98]

# now if I feel like I need the 4th power of some numbers in the list 

get_4th_power = nth_power_maker(4)

for x in my_numbers_list:
    print(get_4th_power(x))

get_7th_power = nth_power_maker(7)

Upvotes: 2

Corentin Pane
Corentin Pane

Reputation: 4943

You can define static variables inside functions, almost like you did:

def GetNthPower(x) :
    return x ** GetNthPower.n

GetNthPower.n = 3

print(GetNthPower(2)) #8

Make sure to initialize correctly your GetNthPower.n before first use though.

If you're worried about initialization, you could go for this version which uses a default value 1:

def GetNthPower(x) :
    return x ** (GetNthPower.n if hasattr(GetNthPower, "n") else 1)

I think it would still be better for you to write a function that takes two arguments, or use the predefined ** operator.

Upvotes: 3

Related Questions