Calculate Probability In Python