Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Vẽ phân phối Gamma với tham số alpha, beta trong Python Matplotlib

Để vẽ hàm mật độ xác suất (PDF) của phân phối Gamma với tham số shape (alpha) và scale (beta), ta dùng scipy.stats.gamma.pdf().

Các bước

  • Thiết lập figure size và autolayout.
  • Tạo mảng x bằng np.linspace, tính y = gamma.pdf(x, a=alpha, scale=beta).
  • Vẽ đường cong bằng plot().
  • Thêm chú giải legend().
  • Hiển thị bằng show().

Ví dụ

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

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

x = np.linspace(0, 10, 100)
y = stats.gamma.pdf(x, a=5, scale=0.333)
plt.plot(x, y, "ro-", label=(r'$\alpha=5, \beta=3$'))
plt.legend(loc='upper right')
plt.show()

Kết quả

Vẽ phân phối Gamma với tham số alpha, beta trong Python Matplotlib