Numpy Apply Function To 2d Array