How to get a normal distribution within a range in numpy? [duplicate]

by

The parametrization of truncnorm is complicated, so here is a function that translates the parametrization to something more intuitive:

from scipy.stats import truncnorm

def get_truncated_normal(mean=0, sd=1, low=0, upp=10):
    return truncnorm(
        (low - mean) / sd, (upp - mean) / sd, loc=mean, scale=sd)

How to use it?

  1. Instance the generator with the parameters: mean, standard deviation, and truncation range:

    >>> X = get_truncated_normal(mean=8, sd=2, low=1, upp=10)

  2. Then, you can use X to generate a value:

    >>> X.rvs()
6.0491227353928894

  3. Or, a numpy array with N generated values:

    >>> X.rvs(10)
array([ 7.70231607,  6.7005871 ,  7.15203887,  6.06768994,  7.25153472,
        5.41384242,  7.75200702,  5.5725888 ,  7.38512757,  7.47567455])

A Visual Example

Here is the plot of three different truncated normal distributions:

X1 = get_truncated_normal(mean=2, sd=1, low=1, upp=10)
X2 = get_truncated_normal(mean=5.5, sd=1, low=1, upp=10)
X3 = get_truncated_normal(mean=8, sd=1, low=1, upp=10)

import matplotlib.pyplot as plt
fig, ax = plt.subplots(3, sharex=True)
ax[0].hist(X1.rvs(10000), normed=True)
ax[1].hist(X2.rvs(10000), normed=True)
ax[2].hist(X3.rvs(10000), normed=True)
plt.show()

enter image description here

More Related Contents:

  1. Convert array of indices to 1-hot encoded numpy array
  2. Consistently create same random numpy array
  3. How to implement the Softmax function in Python
  4. gradient descent using python and numpy
  5. Convert array of indices to one-hot encoded array in NumPy
  6. Extract upper or lower triangular part of a numpy matrix
  7. How to specify upper and lower limits when using numpy.random.normal
  8. Binary random array with a specific proportion of ones?
  9. Generate a random sample of points distributed on the surface of a unit sphere
  10. Difference between np.random.seed() and np.random.RandomState()
  11. Should I use `random.seed` or `numpy.random.seed` to control random number generation in `scikit-learn`?
  12. Custom transformer for sklearn Pipeline that alters both X and y
  13. Creating same random number sequence in Python, NumPy and R
  14. memory issues when transforming np.array using to_categorical
  15. Generating Discrete random variables with specified weights using SciPy or NumPy
  16. Better way to shuffle two numpy arrays in unison
  17. Shuffling NumPy array along a given axis
  18. scikit-learn random state in splitting dataset
  19. How to generate a random normal distribution of integers
  20. Weighted random sample without replacement in python
  21. Neural network XOR gate not learning
  22. How to implement the ReLU function in Numpy
  23. ModuleNotFoundError: No module named ‘numpy.testing.nosetester’
  24. Numpy Broadcast to perform euclidean distance vectorized
  25. Vectorizing `numpy.random.choice` for given 2D array of probabilities along an axis
  26. Working with floating point NumPy arrays for comparison and related operations
  27. Generate random locations within a triangular domain
  28. ValueError: x and y must be the same size
  29. Generate random array of floats between a range
  30. Python: Choose random line from file, then delete that line

Leave a Comment