Array Numpy Apply Along Axis With Error Valueerror Could Not Broadcast Input Array From Shape 2