Numpy Indexing Using Array Formulas