numpy 求函数u=nxy的和，其中n =[ 0 to N] ;x和y是一个数组

vnzz0bqm 于 2023-03-30 发布在其他

关注(0)|答案(2)|浏览(171)

我想用Python写一段代码来解决一个函数，如下所示：

u = sum(n*x*y) for n=[0 to N]

假设x = [1，2，3，4，5]，y =[6，7，8，9，10]，n= [1，2，..100]
我希望输出如下所示：
u = [u0，u1，u2，u3..];其中u 0 = sum（[nn*x[0]*y[0] for nn in n]），并且对于u1，u2…
想到这样做：

u = []
For i in n:
    j = sum([i*x*y])
    u.append(j)

当然现在我遇到的问题是idk我如何在循环中定义x和y。可能需要使用另一个for循环或while循环，或者一些if/else，但由于某种原因，我无法将我的思想缠绕在它周围。对编码来说很新，因此任何帮助都将不胜感激。

numpy

来源：https://stackoverflow.com/questions/75849425/finding-sum-of-a-function-u-nxy-for-n-0-to-n-x-and-y-are-some-array

2条答案

按热度按时间

gkl3eglg1#

我相信您想要：

uu = sum( nn*(sum(xx*yy) for xx,yy in zip(x,y)) for nn in n )

类似但更好：

uu = sum( nn*(sum(x[i]*y[i]) for i in range(len(x))) for nn in n )

后续

u = [sum([nn*xx*yy for nn in n]) for xx,yy in zip(x, y)]

赞(0）回复(0）举报 2023-03-30

7vux5j2d2#

您是否希望：

x = [1,2,3,4,5]
y = [6,7,8,9,10]
n = [1,2,100]

u = [sum([nn*x[i]*y[i] for nn in n]) for i in range(len(x))]

# or
u = [sum([nn*x_i*y_i for nn in n]) for x_i, y_i in zip(x, y)]

输出：

[618, 1442, 2472, 3708, 5150]

numpy版本：

x = np.array([1,2,3,4,5])
y = np.array([6,7,8,9,10])
n = np.array([1,2,100])

u = ((x*y)*n[:,None]).sum(0)

输出：

array([ 618, 1442, 2472, 3708, 5150])

赞(0）回复(0）举报 2023-03-30

我来回答

numpy 求函数u=nxy的和，其中n =[ 0 to N] ;x和y是一个数组

2条答案

后续

相关问题

热门标签

最新问答

numpy 求函数u=n*x*y的和，其中n =[ 0 to N] ;x和y是一个数组

2条答案

后续

相关问题

热门标签

最新问答

numpy 求函数u=nxy的和，其中n =[ 0 to N] ;x和y是一个数组