Для массива numpy 2d, как найти все пары смежных элементов?

Question

Пожалуйста, помогите мне. Я хочу спросить о numpy матрице (2D массив).

У меня есть двумерный массив x, для которого все элементы имеют 1 или 0 в качестве значения. Форма x не более (1000, 1000).

Позвольте мне определить, что: два различных элемента x называются смежной парой, если их индексы (строки и столбцы соответственно) отличаются 1 в наиболее; и оба имеют 1 в качестве значений.

Я хочу знать, как найти все соседние пары на A.

Нужно ли использовать a для l oop?

Большое спасибо заранее.

Answer 1

Я думаю, что требования могут быть выполнены без явных циклов в numpy.

import numpy as np 

np.random.seed(1234)  # Make random array reproduceable

arr = np.random.randint( 0, 2, size = (10,10))

leftshifted = arr[ :, 1:] # Shift arr 1 col left, shape = (10,  9)
downshifted = arr[ 1: ]   # Shift arr 1 row down, shape = ( 9, 10)

hrows, hcols = np.where( arr[ :, :-1 ] & leftshifted ) 
# arr[ :,:-1 ] => ignore last column for the comparison
# returns rows and columns where arr and leftshifted = 1
# i.e. where two adjacent columns in a row are 1

vrows, vcols = np.where( arr[ :-1 ] & downshifted )
# arr[ :-1 ] => ignore last row for the comparison
# returns rows and columns where arr and downshifted = 1
# i.e. where two adjacent rows in a column are 1

print(arr, '\n')
# [[1 1 0 1 0 0 0 1 1 1]
#  [1 1 0 0 1 0 0 0 0 0]
#  [0 0 0 0 1 0 1 1 0 0]
#  [1 0 0 1 0 1 0 0 0 1]
#  [1 1 0 1 1 0 1 0 1 0]
#  [1 1 1 1 0 1 0 1 1 0]
#  [0 1 0 0 1 1 1 0 0 0]
#  [1 1 1 1 1 1 1 0 1 0]
#  [1 0 1 0 0 0 0 0 0 0]
#  [0 1 1 1 0 1 0 0 1 1]]

print('Row indices  :', hrows)
print('Col start ix :', hcols)
print('Col end ix   :', hcols+1)

# Row indices  : [0 0 0 1 2 4 4 5 5 5 5 6 6 7 7 7 7 7 7 9 9 9]
# Col start ix : [0 7 8 0 6 0 3 0 1 2 7 4 5 0 1 2 3 4 5 1 2 8]
# Col end ix   : [1 8 9 1 7 1 4 1 2 3 8 5 6 1 2 3 4 5 6 2 3 9]

print('\nStart Row:', vrows, '\nEnd Row  :',vrows+1, '\nColumn   :', vcols)

# Start Row: [0 0 1 3 3 4 4 4 4 5 5 6 6 6 6 7 7 8] 
# End Row  : [1 1 2 4 4 5 5 5 5 6 6 7 7 7 7 8 8 9] 
# Column   : [0 1 4 0 3 0 1 3 8 1 5 1 4 5 6 0 2 2]

Может ли элемент быть в более чем одной паре? В вышесказанном это может быть. Два ли диагонально соприкасающихся элемента считаются парой? Если это так, то потребуется также массив shiftted_left_and_down.