Numpy Compare Array Shape Python Code Runner Example