Numpy Find Subarray In Array Definition Python