General PMF method

[pelegm]

A general PMF method for random variables should look as follows:

def indicator(self, k):
    """ Return the indicator function of X=k. """
    def ind(w, k=k, func=self.func):
        return 1 if func(w) == k else 0
    return ind

def pmf(self, k):
    return self.pspace.integrate(self.indicator(k))

[pelegm]

Perhaps the ind function should simply return func(w) == k.