Python求和

Question

我是一个非常新的python用户。
我正在尝试计算求和级数。

Equation

我有一张表格，其中包含a_(ij)和p = 10的值（见下文）。

data = [["a_{ij}", "j=0", "j=1", "j=2", "j=3"],
["i=1", 4.3, 8.3, 2.9, `1.3],
["i=2", 1.4, 4.7, 6,5, 3.2]]

---

a_(ij)  j = 0  j = 1  j = 2  j = 3
i = 1    4.3    8.3    2.9     1.3
i = 2    1.4    4.7    6.5     3.2
.
.
i = 20   8.34   2.13  7.93   6.322

不是为y的每个值分别编写代码。

# For i = 1
y_1 = 4.3*10**0 + 8.3*10**1 + 2.9*10**2 + 1.3*10**3

有没有更简单的节省时间的方法？

Answer 1

将a_（ij）存储为矩阵。将其设为2D数组，或使用numpy库并将其设为numpy矩阵。

二维数组示例：

a = [[4.3, 8.3, 2.9, 1.3],
     [1.4, 4.7, 6.5, 3.2],
     # you've indicated more lines will go here
     [8.34, 2.13, 7.93, 6.322]]

如果a是可以通过函数生成的东西，我建议您以这种方式生成2D数组，而不是手动输入值。

然后您可以将公式实现为函数：

def y(i):
    p = 10
    total = 0
    for j in range(1,4):
        total += a[i][j] * p**j
    return total

现在您可以通过调用以下函数来获取y的值，以获取任意i值。 y(1)将给出i = 1的y值。

您还可以为y生成一个数组（如果您经常重复使用相同的y值，那么如果将其缓存在数组中而不是每次都重新计算y，则代码将运行得更快）。

Answer 2

这是一个干净的解决方案：

>>> a = [
...     [4.3, 8.3, 2.9, 1.3],
...     [1.4, 4.7, 6,5, 3.2]
... ]
>>> p = 10.0
>>> import operator
>>> [sum(map(operator.mul, i, map(p.__pow__, range(len(i))))) for i in a]
[1677.3, 37648.4]

如果您使用的是python 2.7，请用zip替换izip并从izip导入itertools。

这是您发布以进行比较的手动计算的结果：

>>> 4.3*10**0 + 8.3*10**1 + 2.9*10**2 + 1.3*10**3
1677.3