Numpy Compare Array Shape Python Code For Game