简体   繁体   中英

Modifying a global variable inside a function

I have defined the following function:

def GMM(s1, s2, s3, s4, s5, a):
    """The GMM objective function.

    Arguments
    ---------
        si: float
            standard deviations of preference distribution

        a: float
            marginal utility of residutal income

    Paramters
    ---------
        Px: array (1,ns) 
            projector onto nonprice characteristic space

        xk, z: arrays (J, 5) and (J, 12)
            nonprice char. and instruments

        invW: array (12, 12) 
            GMM weight matrix

    Returns
    -------
        float."""
    delta = invert(s1, s2, s3, s4, s5, a, delta0) # Invert market shares to get mean utility
    bmean = np.dot(Px, delta)                     # Project delta onto charancteristic space
    xihat = delta - np.dot(xk, bmean)             # Compute implied unobservable prod. quality
    temp1 = np.dot(xihat.T, z)
    if np.any(np.isnan(delta)) == True:
        value = 1e+10
    else:
        value = np.dot(np.dot(temp1, invW), temp1.T)
    return np.sqrt(value)

My question pertains to the variable delta bound inside of the function. Outside of the function I will set the initial value of delta0 . Now, ultimately I will minimize this function. What I would like to have happen is that each time the function GMM evaluates, delta from the previous evaluation is used as the new delta0 . I tried defining delta0 as a global variable, but it did not seem to work... likely this was my error though. Although, I have read here that generally this is a bad approach. Any suggestions?

There are multiple ways to achieve what you want. delta is saved across function calls in the following examples.

1- Class

class Example:
  def __init__(self, value):
    self.delta = value
  def gmm(self):
    self.delta += 1
    return self.delta

e = Example(0)
print e.gmm()

2- Generator

def gmm():
  delta = 0
  while True:
    delta += 1
    yield delta

for v in gmm():
  print v

3- Function attribute

def gmm():
  gmm.delta += 1
  return delta
gmm.delta = 0

4- Global variable (discouraged as you said):

delta = 0
def gmm():
  global delta
  delta += 1
  return delta

etc...

globalVariable = 0

def test():
    global globalVariable
    globalVariable = 10

test()
print globalVariable

You can edit a global variable in this way.

When faced with this, a common kludge I use is stuff an object into a module, which then puts it into a namespace accessible by everything in the program. It is a big kludge, but I find it removes any ambiguity about what's global. For standalone stuff I put it into os , if it's an entire project I'll generally create an empty python file called my_globals and import it, ie

import my_globals
my_globals.thing = "rawp"
def func():
    my_globals.thing = "test"
func()
print my_globals.thing # "test"

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM