Using Function in a For loop

[vp1989]

I simplified the code for the readers.

import numpy as np
import sys
import math
import copy
row_matrix=9   #row
coulomn_matrix=9
tstep=10 #number of one iteration  
interval=5 #lenghth of one interval
max_matrix=np.zeros((1,row_matrix,coulomn_matrix)) #maximum of matrix
maxm=[] # used for copy max_matrix
mat=[]
matrix=[]
save_matrix=[]
for N in range(0,5):
   if (N==0):   # maxm initiation
       maxm=zeros(row_matrix,Dim)
   else: 
       def max():     ## A function to calculate C
        for i in range(0,interval):  
            if(i%interval==0):   ###### getting a max in a interval 
                for j in range(0,row_matrix):
                    for k in range(0,coulomn_matrix):
                        max_matrix[0][j][k]=mat[i][j][k]     ##### the K is comeing from the main body 
            elif(i%interval!=0):
                for j in range(0,row_matrix):
                    for k in range(0,coulomn_matrix):
                        if (mat[i][j][k]>max_matrix[0][j][k]): ##### getting max value 
                            max_matrix[0][j][k]=mat[i][j][k]
            if((i+1)%interval==0):   #### saving max value for the current interval 
                maxm=P.copy()
            return maxm
###### get maxm for each interval 
    for nstep in np.arange(tstep):
        V=[]
        for i in range(0,row_matrix):
            U=[]
            for j in range(0,row_matrix):
                m=18+5*i*nstep/2-6*j*i*nstep*N*N  #calculation of value of indicies
                U.append(m)                       
            V.append(U)
        matrix.append(V)### getting the matrix 
    mat=np.array(matrix)   ### getting arrays to use in functioin max 
save_matrix.append(matrix) ###saving matrix at each interval