如果一维数组等于其相邻数组的算术平均值，则为一维数组的计数

Question

我想找出一个数组中的数字，它等于它们的近邻的算术平均值。这里的转折点是，如果[I-1]或[I+1]不存在，则使用0

例如：

a =  [2,4,6,8,2,5]
a[0] = (0 + 4)/2 = 2 ----count 1
a[1] = (2 + 6)/2 = 4---- count 2
a[5] = (2 + 0)/2 != 5 ----no count

如果不是，则计数应为0。我已经编写了下面的代码，但它不能适用于所有的数组，就像这一个。我怎么做

a = [100, 110, 120, 130, 140, 150, 160, 170, 180, 190]

我的代码：

def countMeans(a):
    count = 0
    for i in range(1, len(a)-1):
        if a[i-1]+ a[i+1] is 2 * a[i]:
            count += 1
    return count

Answer 1

根据您的新更新，请使用以下代码：

def countMeans(a):
    count = 0
    if len(a) > 1:
        for i in range(0, len(a)):
            val1 = 0 if i == 0 else a[i-1]
            try:
                val2 = a[i+1]
            except:
                val2 = 0
            if val1 + val2 == 2 * a[i]:
                count += 1
    else:
        count = 1 if a[0] == 0 else 0
    return count