Numpy Array Index Conditional Probability Examples Sports