Discrete random variables in Python made easy