Início / coding / Perguntas / 77426326
Accepted
FreelanceConsultant
Asked: 2023-11-05 22:33:51 +0800 CST

Potência numpy com array como expoente

  • 772

Eu tenho um código Python contendo as seguintes linhas:

# Poisson model

lambda_param = 3.0

x_model = numpy.linspace(0, 10, num=11, dtype=int)
y_model = numpy.zeros(10)

index = 0
for x in x_model:
    y_model[index] = numpy.power(lambda_param, x) / math.factorial(x) * numpy.exp(-lambda_param)
    index += 1

Quero escrever isso de uma forma mais pitônica - sem usar um loop for.

Eu tentei isso:

y_model = numpy.power(lambda_param, x_model) / math.factorial(x_model) * numpy.exp(-lambda_param)

Mas isso não pareceu funcionar, eu acho, porque math.factorialnão sabe como operar em um objeto semelhante a um array.

Existe uma maneira de fazer isso?

python

2 respostas

  1. Best Answer

    Este é o pmfcaso de uma distribuição de poisson. Uso scipy.stats.poisson: Observe que para poisson, xdevem ser números inteiros:

    lambda_param = 3.0
x_model = np.arange(10)
y_model = np.zeros(10)

from scipy import stats
stats.poisson(lambda_param).pmf(x_model)
array([0.04978707, 0.14936121, 0.22404181, 0.22404181, 0.16803136,
       0.10081881, 0.05040941, 0.02160403, 0.00810151, 0.0027005 ])
    • 2

  2. Você pode usar scipy.special.factorialcomo uma função fatorial que suporta transmissão em matrizes numpy.

    import numpy as np
import scipy.special

>>> scipy.special.factorial(np.arange(7))
array([  1.,   1.,   2.,   6.,  24., 120., 720.])
    • 1

relate perguntas