python - Python 中两个列表元素的计算求和

Question

我需要对 (y_i-A-B*x_i)^2 从 i=1 到 n 求和。

A和B是固定常数。 y_i 和 x_i 是两个列表的元素。

我尝试使用 sum([m-A-B*n for m,n in zip(Tlog_dist,Tlog_Tb)]) 但我认为对所有可能的数字对进行求和，而不仅仅是 i 'th 与 ith。

我环顾四周，看到了一些带有 for 循环和枚举的东西，但被卡住了，因为我是新手，以前从未使用过循环。

Answer 1

使用numpy作为矢量化解决方案。

import numpy  as np

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

# (y_i-A-B*x_i)^2
res = np.sum((y - A - B*x)**2)

基于列表的解决方案是可能的，但不推荐。