matlab for loop to python loop

Question

I'm trying to change this Matlab for loop:

n =100;
u_lsg = zeros (n ,n) ; 
for i =1:1: n
    for j =1:1: n
        u_lsg(i , j) =( i*h)*(j*h )*(1 -( i*h ))*(1 -( j*h ));
    end
end

into a python loop.

What I've got so far:

n =100
u_lsg = np.zeros((n ,n))
for i in range (1,n+1,1):
    for j in range (1,n+1,1):
        u_lsg[i,j]= (i*h)*(j*h)*(1-(i*h))*(1-(j*h))

But this gives IndexError: index 100 is out of bounds for axis 1 with size 100.

What did I do wrong? And from what I understand there is no end command in python, but then how does one end the loop?

Djaouad · Accepted Answer · 2021-07-31 15:44:03Z

1

Matlab is 1-indexed (array indices start at 1), Python (and NumPy) is 0-indexed (array indices start at 0), so just change the ranges accordingly:

n = 100
u_lsg = np.zeros((n, n))
for i in range(n):
    for j in range(n):
        u_lsg[i, j] = (i*h)*(j*h)*(1-(i*h))*(1-(j*h))

edited Jul 31, 2021 at 15:44

answered Jul 31, 2021 at 15:29

Djaouad

22.8k4 gold badges37 silver badges57 bronze badges

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

Collectives™ on Stack Overflow

matlab for loop to python loop

1 Answer 1

Comments

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Related