Numpy Compare Array Shape Python Code Compiler