Using fancy indexing to select all the models as a given redshift is very suboptimal as it forces a copy for each of the index arrays. Now because the redshift is the last parameter, we can just use a proper slice. The first element is the index of the redshift in the list of redshift (paying attention that this list is in the same order as the computation order), and the step is the number of redshifts.