Блог

Я пытаюсь создать функцию, которая принимает два входа. Один вход-матрица (n*m), а второй-K. K-целочисленное значение. Расстояние между ячейками A[3][2] и A[1][4] есть |1-3| |4-2| = 4 . Ожидаемый результат функции-это количество ячеек с расстоянием между ячейками больше K.

Ячейка здесь-это каждая запись в данной матрице A. Например, [0][0] — это ячейка, и она имеет значение 1 в матрице.

Я создал функцию, подобную этой:

 A = [[1, 0, 0], 
     [0, 0, 0],
     [0, 0, 1],
     [0, 1, 0]]

def findw(K, matrix):
    m_c = matrix.copy()
    result = 0
    for i, j in zip(range(len(matrix)), range(len(m_c))):
        for k, l in zip(range(len(matrix[i])), range(len(m_c[j]))):
            D = abs(i - l)   abs(j - k)
            print(i, k)
            print(j, l)
            print(D)
            
            if D > K:
                result  = 1
    return result

findw(1, A)
 

Результат, который я получил из приведенной выше функции для данной матрицы A с K = 1, равен 9. Но я ожидаю 3. Из вывода я также понял, что для обеих матриц моя функция всегда принимает одно и то же значение , например (0,0) , или (1,0) и т. Д. Смотрите результаты печати ниже.

 findw(1, A)
0 0
0 0
0
0 1
0 1
2
0 2
0 2
4
1 0
1 0
2
1 1
1 1
0
1 2
1 2
2
2 0
2 0
4
2 1
2 1
2
2 2
2 2
0
3 0
3 0
6
3 1
3 1
4
3 2
3 2
2
Out[120]: 9
 

Похоже, что моя функция не повторяется для ячеек, где индексы для обеих матриц различны. Например, matrix[0][0] И. m_c[0][1]

Как я могу решить эту проблему?

Работая в предположении 1 , что вас интересуют только позиции, которые имеют значение, вы можете сначала перечислить эти индексы, а затем перебрать пары таких вещей. itertools-это естественный инструмент для использования здесь:

 from itertools import product, combinations

def D(p,q):
    i,j = p
    k,l = q
    return abs(i-k)   abs(j-l)

def findw(k,matrix):
    m = len(matrix)
    n = len(matrix[0])
    result = 0
    indices = [(i,j) for i,j in product(range(m),range(n)) if matrix[i][j] == 1]
    for p,q in combinations(indices,2):
        d = D(p,q)
        if d > k:
            print(p,q,d)
            result  = 1
    return result

#test:

A = [[1, 0, 0], 
     [0, 0, 0],
     [0, 0, 1],
     [0, 1, 0]]

print(findw(1, A))
 

Выход:

 (0, 0) (2, 2) 4
(0, 0) (3, 1) 4
(2, 2) (3, 1) 2
3