How to check if an array is multidimensional

Question

I am trying to write a code to check if n is the maximum of the array A.

If it is return True if it isn't return False, but I have two conditions: if the value is multidimensional or unidimensional.

I don't know how to write that.

 if [A is unidimentional]:
     maximum=A[0]
     for i in range(A.shape[0]):
         if max(A[i],maximum)==A[i]:
             maximum=A[i]
     if max(n,maximum)!=n:
         return False
     return True
    
     else:
         maximum=A[0][0]
         for i in range(A.shape[0]):
         for j in range(A.shape[1]):
             if max(A[i][j],maximum)==A[i][j]:
                 maximum=A[i][j]
     if max(n,maximum)!=n:
         return False
     return True

How can I write that conditional logic into my script?

Answer 1

def is_multidimensional(A):
    return A.ndim > 1