Numpy Array Index Conditional Probability Examples Basic