python - 我正在尝试遍历多个范围以从另一个范围创建多维 ndarray

Question

我有：

>>> import numpy as np
>>> a = np.arange(25).reshape(5, 5)
>>> a
array([[ 0,  1,  2,  3,  4],
       [ 5,  6,  7,  8,  9],
       [10, 11, 12, 13, 14],
       [15, 16, 17, 18, 19],
       [20, 21, 22, 23, 24]])

如何获得一个数组，它是一个数字和它下面的数字之和：

array([[7,  9,  11,],   # 1+6=7 , 2+7=9 , 3+8=11
       [17, 19, 21,],
       [27, 29, 31,]])

我想通过迭代原始数组来做到这一点。就像是：

b[x,y] = [a[x,y]+a[x+1,y] for x in range(0,3) for y in range(1,4)] #ERROR!

尽管由于语法原因这不起作用。有人可以给我正确的语法吗？我不是专业程序员，对 Python 很陌生。先感谢您。

Answer 1

使用切片来挑选 (1) 除了数组的最后一行之外的所有内容和 (2) 除了数组的第一行之外的所有内容。然后添加它们。

>>> a[:-1,:] + a[1:,:]
array([[ 5,  7,  9, 11, 13],
       [15, 17, 19, 21, 23],
       [25, 27, 29, 31, 33],
       [35, 37, 39, 41, 43]])