How to create a for loop to process a multid-dimensional matrix in Python

Question

Assume I have a four-dimensional matrix A(:, :, :, :). I want to update the matrix by performing some processing on it. The pseudo codes are presented as follows:

for ii = 1:m for jj = 1:n A = myFunction(A(:,:,jj,ii)) end end

To implement the for-loop processing in Python:

for ii in range(m): for jj in range(n): A = myFunction(A[:,:,jj,ii])

Is that correct?

for multi-dimension matrix and these kind of loops, ndarray from numpy would be best choice. — rnbguy
– rnbguy, Commented Oct 26, 2016 at 22:41
You mean the code structure is valid in Python? It seems I need to make the index range from 0 to max-1 and delete the two 'end' . I am new to Python programming. — jwm
– jwm, Commented Oct 26, 2016 at 22:42

Pavel · Accepted Answer · 2016-10-26 23:17:41Z

If you have 4-dimensional matrix, you should use 4 indexes:

for i in range(m): for j in range(n): for k in range(p): for l in range(q): myFunction(A[i,j,k,l])

For example:

A = [[[[6,1],[4,3]],[[4,8],[0,9]]],[[[1,5],[3,9]],[[5,5],[2,7]]]] s = 0 for i in range(2): for j in range(2): for k in range(2): for l in range(2): s += A[i][j][k][l] print(s)

Collectives™ on Stack Overflow

How to create a for loop to process a multid-dimensional matrix in Python

1 Answer 1

Comments

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Related