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

发布时间：2020-12-20 11:31:15 所属栏目：Python 来源：网络整理

导读：我有： import numpy as np a = np.arange(25).reshape(5,5) aarray([[ 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,],# 1+6=7,2+7=9,3+8=11 [17,19,[27,29,31,]]) 我想

我有：

>>> 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,],# 1+6=7,2+7=9,3+8=11
       [17,19,[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,13],23],[25,27,33],[35,37,39,41,43]])

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!