diff --git a/quantecon/random/utilities.py b/quantecon/random/utilities.py index f1b53f5d..9838b2e1 100644 --- a/quantecon/random/utilities.py +++ b/quantecon/random/utilities.py @@ -200,9 +200,7 @@ def draw(cdf, size=None): """ if isinstance(size, int): rs = np.random.random(size) - out = np.empty(size, dtype=np.int_) - for i in range(size): - out[i] = searchsorted(cdf, rs[i]) + out = np.searchsorted(cdf, rs, side='right') return out else: r = np.random.random()