Numpy Array Index Conditional Probability Definition