python – 我试图迭代多个范围来从另一个范围创建一个多维的ndarray

2024年1月5日 221次阅读

我有：

>>> 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.先感谢您.

最佳答案 使用切片挑选(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]])
点赞