python - 根据python中另一个数组的条目计算数组中的新条目

Question

我有一个问题，如何在循环另一个数组时“调用”数组中的特定单元格。假设有一个数组a:

    a = [[a1 a2 a3],[b1 b2 b3]]

和一个数组 b:

    b = [[c1 c2] , [d1 d2]]

现在，我想使用数组 a 中的信息重新计算数组 b 中的值。具体来说，数组 b 的每个值都必须通过与数组 a 中给出的边界之间的高斯函数积分相乘来重新计算。但为了简单起见，让我们忘记积分，并假设需要以下形式的简单计算:

    c1 = c1 * (a2-a1) ; c2 = c2 * (a3 - a2) and so on,

使用索引，它可能看起来像:

    b[i,j] = b[i,j] * (a[i, j+1] - a[i,j])

谁能告诉我如何解决这个问题？非常感谢并致以最诚挚的问候，马克

Answer 1

您可以使用zip嵌套列表理解中的函数:

>>> [[k*(v[1]-v[0]) for k,v in zip(v,zip(s,s[1:]))] for s,v in zip(a,b)]

zip(s,s[1:]) 将为您提供所需的元素对，例如:

>>> s =[4, 5, 6]
>>> zip(s,s[1:])
[(4, 5), (5, 6)]

演示:

>>> b =[[7, 8], [6, 0]]
>>> a = [[1,5,3],[4 ,0 ,6]]
>>> [[k*(v[1]-v[0]) for k,v in zip(v,zip(s,s[1:]))] for s,v in zip(a,b)]
[[28, -16], [-24, 0]]