Numpy Array Index Conditional Probability Examples