Updating a particular element of an array in python functions

Question

I have a section of code that I am using to calculate magnetic fields and magnetic vector potentials. I do various things to calculate an inverse curl, but I'm getting stuck on a namespace issue in python and I'm not sure how to move forward.

The code (shortened a little) works like this

def main():
    # define all my arrays 
    Ax = np.zeros(nx)
    # call a function to update a particular component of the Ax 
    index = 50
    update_A(Ax[index])
    return Ax 

def update_A(Axmm):
   # do a calculation to get this new component
   Axmm = some function

The update_A function is calculating the value as expected, but then isn't overwriting the element of Ax like I want it to. How can I get it to do the calculation and update the element of Ax that I've given it?

Ax[index] = calc_ new_A(Ax[index]) most types in python are immutable, so you pass to function not "array cell", but just copy of array element and assignment has no effect to array — NobbyNobbs
– NobbyNobbs, Commented Mar 11, 2021 at 11:09

user12983953 · Accepted Answer · 2021-03-11 11:08:34Z

1

#Pass and Index to your update function.

def update_A(Axmm, index):
   # do a calculation to get this new component
   Axmm[index] = some function

#Let me know if that's not working the way you meant.

answered Mar 11, 2021 at 11:08

user12983953

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Updating a particular element of an array in python functions

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related