python - python中的求和求和

Question

给出z = np.linspace(1,10,100)

计算 z^k * exp((-z^2)/2) 中 z 的所有值的总和

import numpy as np
import math

def calc_Summation1(z, k):
    ans = 0.0
    for i in range(0, len(z)):`
        ans += math.pow(z[i], k) * math.exp(math.pow(-z[i], 2) / 2)
    return ans

def calc_Summation2(z,k):
     part1 = z**k
     part2 = math.exp(-z**2 / 2)
     return np.dot(part1, part2.transpose())

有人能告诉我 calc_Summation1 和 calc_Summation2 有什么问题吗？

Answer 1

我认为这可能是您正在寻找的:

sum(z_i**k * math.exp(-z_i**2 / 2) for z_i in z)