Numpy Array Index Conditional Probability Practice