сначала вам нужно импортировать sqrt из квадратов математической библиотеки, можно либо рассчитать, умножив значение на себя (r2-r1) (r2-r1), либо используя pow из математической библиотеки.(r2-r1) ^ 2 не работает.
Строго говоря, A - это не массив, а список (строки) списков (столбцов).Но вы все равно можете считать это своего рода массивом.Вы получаете его значения, используя два индекса A [row_index] [column_index]
from math import sqrt, pow
A = [[206,205,204,190,208],
[190,194,206,197,203],
[196,196,205,201,193],
[194,199,199,206,205],
[192,196,195,201,193],
[194,199,200,200,205],
[196,196,195,200,193]]
W = 20
def distance(r1, c1, r2, c2):
# get a values for point 1 and 2
a1 = A[r1][c1]
a2 = A[r2][c2]
# calculate the distance
d = sqrt(pow(W*(r2-r1), 2) + pow(W*(c2-c1), 2) + pow(a2-a1, 2))
return d
print(distance(0,0,4,4))
>>> 113.88