Numpy如何在不使用for循环的情况下对表内容进行移动(增长)求和?

gmxoilav  于 2022-12-23  发布在  其他
关注(0)|答案(1)|浏览(107)

由于英语不是我的第一语言,我很难简单地解释我在这个主题中试图解决的问题,因此我很抱歉。
因此,我不想用简单的语言来解释,而是要给予一个例子。
假设我们有一个示例化的数组,如下所示:

weight = np.arange(1, (n + 1)).astype('float64')

所以数组看起来像这样:

[ 1.  2.  3.  4.  5.  6.  7.  8.  9. 10.]

现在我想做的是有一个移动和数组(我们称之为norm),对数组norm求和,运算如下所示:

index, norm(new array), weight, operation
  0        1               1     0+1 = 1
  1        3               2     0+1+2 = 3
  2        6               3     0+1+2+3 = 6
  3        10              4     0+1+2+3+4 = 10
  .         .              .      .
  .         .              .      .
  .         .              .      .
  9        55              10    0+1+2+3+...+10 = 55

我希望这是可以理解的。我如何在不循环遍历权重数组的情况下获得这个结果?

相关问题