Numpy Array Index Conditional Distribution