Функция плотности вероятности в SciPy ведет себя не так, как ожидалось - PullRequest
Новая
       56

Функция плотности вероятности в SciPy ведет себя не так, как ожидалось

0 голосов
/ 31 марта 2019

Я пытаюсь построить нормальную кривую распределения, используя Python.Сначала я сделал это вручную, используя обычную функцию плотности вероятности, а затем обнаружил, что в модуле scipy under stats есть функция выхода pdf.Однако результаты, которые я получаю, совершенно разные.

Ниже приведен пример, который я попробовал: 

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

mean = 5
std_dev = 2
num_dist = 50

# Draw random samples from a normal (Gaussion) distribution
normalDist_dataset = np.random.normal(mean, std_dev, num_dist)

# Sort these values.
normalDist_dataset = sorted(normalDist_dataset)

# Create the bins and histogram
plt.figure(figsize=(15,7))
count, bins, ignored = plt.hist(normalDist_dataset, num_dist, density=True)

new_mean = np.mean(normalDist_dataset)
new_std = np.std(normalDist_dataset)

normal_curve1 = stats.norm.pdf(normalDist_dataset, new_mean, new_std)
normal_curve2 = (1/(new_std *np.sqrt(2*np.pi))) * (np.exp(-(bins - new_mean)**2 / (2 * new_std**2)))

plt.plot(normalDist_dataset, normal_curve1, linewidth=4, linestyle='dashed')
plt.plot(bins, normal_curve2, linewidth=4, color='y')

Результат показывает, как две кривые, которые я получаю, сильно отличаются друг от друга.

enter image description here

Я предполагаю, что это как-то связано с bins или pdf ведет себя иначе, чем обычная формула.Я использовал одинаковое и новое среднее значение и стандартное отклонение для обоих графиков.Итак, как мне изменить мой код, чтобы он соответствовал тому, что делает stats.norm.pdf?

Я пока не знаю, какая кривая верна.

1 Ответ

2 голосов
/ 31 марта 2019

Функция plot просто соединяет точки с отрезками. В ваших корзинах недостаточно точек, чтобы показать плавную кривую. Возможное решение: 

....
normal_curve1 = stats.norm.pdf(normalDist_dataset, new_mean, new_std)
bins = normalDist_dataset # Add this line
normal_curve2 = (1/(new_std *np.sqrt(2*np.pi))) * (np.exp(-(bins - new_mean)**2 / (2 * new_std**2)))
....
...