减去相邻的列表元素，直到它们不能

Question

行为：

我试图将它与我的项目集成，但是我只能以长方法实现它。 请推荐一个较短的。

Answer 1

使用 numpy 逐项减去

import numpy as np

def func(sequence):
    if len(sequence) == 1:
        return sequence[0]
    arr1 = np.array(sequence[:-1])
    arr2 = np.array(sequence[1:])
    print(arr2-arr1)
    return func(list(arr2 - arr1))

print(func([1, 4, 9, 16, 25]))
print(func([1, 8, 27, 64, 125, 216]))

[3 5 7 9]
[2 2 2]
[0 0]
[0]
0
[ 7 19 37 61 91]
[12 18 24 30]
[6 6 6]
[0 0]
[0]
0