总结邻居数

Question

我正在尝试遍历列表中的所有数字，并且我想将每个数字与他的邻居相加。

    DataList=[2,7,18,22]
    total=0
    for number in DataList:
        total+=number
    print(DataList[0] + DataList[1])
    print(DataList[0] + DataList[1]+DataList[2])
    print(DataList[1] + DataList[2]+DataList[3])
    print(DataList[2] + DataList[3])

#Output from code above
9
27
47
40

正如您从上面的代码中看到的那样，我已经提出了一些解决方案，但是这个解决方案并不好（实际上我有非常大的示例）因为我用 print 输入了所有命令，所以任何人都可以帮助我如何解决这个问题并进行自动循环和对所有邻居数求和？

Answer 1

只需将if-elif-else与循环混合使用：

DataList=[2,7,18,22]
for i in range(len(DataList)):
    if i == 0:
        print(DataList[i] + DataList[i+1])
    elif i == len(DataList)-1:
        print(DataList[i-1] + DataList[i])
    else:
        print(DataList[i-1] + DataList[i] + DataList[i+1])

Answer 2

itertools 中有一个 function ，它是累积的，它实际上可以按照您的意愿运行。 这是 python 文档，您可以在这里找到累积的https://docs.python.org/3/library/itertools.ZA7635FDC70D2EFC26

Answer 3

DataList=[2,7,18,22]
xp=[items for item in [[sum(DataList[:x+1]),sum(DataList[x:])]for x in range(len(DataList))] for items in item]
print(xp)

Answer 4

我认为这是您打算拥有的：

totalNum = len(DataList)
totalSum = 0
sumArray = [0]*totalNum

for i in range(totalNum):
    sumArray[i] += DataList[i]
    if i+1 < totalNum:
        sumArray[i] += DataList[i+1]
    if i-1 > 0:
        sumArray[i] += DataList[i-1]
for i in range(totalNum):
    print(sumArray[i])

Answer 5

一种“聪明”的方法是创建三个版本的列表，并附加0 ，然后将它们相加。

DataList = [ 2, 7, 18, 22 ]

l = [ 0, 0 ] + DataList
c = [ 0 ] + DataList + [ 0 ]
r = DataList + [ 0, 0 ]

sums = [ sum( n ) for n in zip( l, c, r ) ]

然后，如果您不想要它们，您可以将元素从末端切掉。

使用这种方法，您还可以使用numpy来加快大型列表的计算速度。

sums = np.sum( np.array( [ l, c, r ] ), axis = 0 )

对于不同大小的列表，您可以看到numpy收益（注意 y 轴是对数刻度）