Случайный массив в некоторых границах - PullRequest
1 голос
/ 10 апреля 2019

У меня есть массив params с ошибками e_params и границы, что массив может быть params_bounds:

params = [0.2, 0.2]
e_params = [0.1, 0.05]
params_bounds = [(0.0, 1.0), (0.0, 1.0)]

Я хочу нарисовать случайную гауссову реализацию params следующим образом:

import numpy as np
params_mc = np.random.normal(params, e_params)

Есть ли способ убедиться, что результат params_mc находится в пределах верхней и нижней границ, указанных в params_bounds?

Спасибо залюбая помощь здесь.

Ответы [ 3 ]

3 голосов
/ 10 апреля 2019

Вы можете использовать numpy.clip, чтобы обрезать значения в заданных пределах.Сначала сгенерируйте массивы необходимых вам минимумов и максимумов, например:

>>> lower_bound = numpy.asarray(param_bounds)[:, 0]
>>> upper_bound = numpy.asarray(param_bounds)[:, 1]

Теперь закрепите ваш результат:

>>> numpy.clip(params_mc, lower_bound, upper_bound)

(непроверенный код, ваш пробег может отличаться)

2 голосов
/ 10 апреля 2019

Возможно, вы ищете усеченное нормальное распределение .Использование scipy.stats.truncnorm,

import numpy as np
import scipy.stats as stats
import matplotlib.pyplot as plt

lower, upper = (0.0, 0.0), (1.0, 1.0)
mu, sigma = np.array([0.2, 0.2]), np.array([0.1, 0.05])
X = stats.truncnorm(
    (lower - mu) / sigma, (upper - mu) / sigma, loc=mu, scale=sigma)
data = X.rvs((10000, 2))

fig, ax = plt.subplots()
ax.hist(data[:, 0], density=True, alpha=0.5, bins=20)
ax.hist(data[:, 1], density=True, alpha=0.5, bins=20)
plt.show()

дает

enter image description here


Вот еще один способ визуализациипример.Код в основном взят из галереи matplotlib :

import numpy as np
import scipy.stats as stats
import matplotlib.pyplot as plt
import matplotlib.ticker as mticker

lower, upper = (0.0, 0.0), (1.0, 1.0)
mu, sigma = np.array([0.2, 0.2]), np.array([0.1, 0.05])
X = stats.truncnorm(
    (lower - mu) / sigma, (upper - mu) / sigma, loc=mu, scale=sigma)
data = X.rvs((10000, 2))
x, y = data.T

nullfmt = mticker.NullFormatter()         # no labels

# definitions for the axes
left, width = 0.1, 0.65
bottom, height = 0.1, 0.65
bottom_h = left_h = left + width + 0.02

rect_scatter = [left, bottom, width, height]
rect_histx = [left, bottom_h, width, 0.2]
rect_histy = [left_h, bottom, 0.2, height]

# start with a rectangular Figure
plt.figure(1, figsize=(8, 8))

axScatter = plt.axes(rect_scatter)
axHistx = plt.axes(rect_histx)
axHisty = plt.axes(rect_histy)

# no labels
axHistx.xaxis.set_major_formatter(nullfmt)
axHisty.yaxis.set_major_formatter(nullfmt)

# the scatter plot:
axScatter.scatter(x, y)

axScatter.set_xlim((-0.1, 0.7))
axScatter.set_ylim((-0.1, 0.5))

bins = 20
axHistx.hist(x, bins=bins)
axHisty.hist(y, bins=bins, orientation='horizontal')

axHistx.set_xlim(axScatter.get_xlim())
axHisty.set_ylim(axScatter.get_ylim())

plt.show()

enter image description here

0 голосов
/ 10 апреля 2019

Просто короткая идея, вы можете использовать np.clip (), чтобы сделать это довольно легко!

params_bounds = [np.clip(params_mc[i], params_bounds[i][0],params_bounds[i][1]) for i in range(len(params_mc))]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...