Как вывести индекс определённого значения в многомерном массиве в Python?

Question

[nuclear_skillet21011]

import numpy as np

def find_index_iin_matrx(grid, row, col): # [row] - ряд, [col] - столбец, [ grid ] - многомерный массив( матрица 5х5 )
					
	mass_el_1 = []
	grid1 = np.array(grid)
	
	for el in grid1.flat: # атрибут flat (numpy) используется при для обработки массива как одномерного
		if el == 1:
			ac = grid1.index(el) # здесь ошибка AttributeError: 'numpy.ndarray' object has no attribute 'index'
			mass_el_1.append(ac)
	print(masvie_index)

Сегодня впервые столкнулся с многомерным массивом в python. Задача тривиальна: вычислить индексы всех не нулевых значений в матрице. И затем проверить какие из значений являются "соседними":

((1, 0, 0, 1, 0),
(0, 1, 0, 0, 0),
(0, 0, 1, 0, 1),
(1, 0, 0, 0, 0),
(0, 0, 1, 0, 0),)

у единицы в второй строке, втором столбце: два соседа с индексами (1,1) и (3,3). Подскажите при помощи какого кода их можно определить?
Спасибо за внимание.

Answer 1

[nirvimel]

1. У массива numpy нет метода index().
2. Вывести все индексы всех ненулевых элементов grid1:

import numpy


def print_all_non_zero_values(arg):
    print ((numpy.array(arg) > 0).astype(numpy.intc))

3. Подсчет ненулевых соседей:

import numpy


def count_neighbours(source):
    grid1 = numpy.array(source) > 0
    x_size, y_size = grid1.shape
    grid2 = numpy.zeros((x_size + 2, y_size + 2), dtype=numpy.intc)
    grid2[ :-2,  :-2] += grid1
    grid2[1:-1,  :-2] += grid1
    grid2[2:  ,  :-2] += grid1
    grid2[ :-2, 1:-1] += grid1
    grid2[1:-1, 1:-1] += grid1
    grid2[2:  , 1:-1] += grid1
    grid2[ :-2, 2:  ] += grid1
    grid2[1:-1, 2:  ] += grid1
    grid2[2:  , 2:  ] += grid1
    return (grid2[1:-1, 1:-1] - 1) * grid1

Тест:

count_neighbours(((0, 0, 1, 0),
                  (0, 1, 1, 0),
                  (0, 1, 1, 0),
                  (0, 1, 0, 0),))

Результат:

array([[0, 0, 2, 0],
       [0, 4, 4, 0],
       [0, 4, 4, 0],
       [0, 2, 0, 0]], dtype=int32)

Comments

[nirvimel]

nuclear_skillet21011: Про что именно прочитать?

[nuclear_skillet21011]

Огромное спасибо! index() втыкал уже от без исходности, не подскажите где (в офф документации или в прочих источниках ) можно прочитать про это по подробнее а то я искал не нашёл был бы очень благодарен спасибо ещё раз!

[nuclear_skillet21011]

про поиск индексов (или про использование numpy в картографических целях)

[nuclear_skillet21011]

чёрт куда делся мой коммент?

[nirvimel]

nuclear_skillet21011: Я не понимаю что значит поиск индекса без относительно конкретной задачи. По принципам индексации в Numpy есть официальная документация - https://docs.scipy.org/doc/numpy/reference/arrays....
Использование Numpy в контексте какой-то прикладной области - слишком общий вопрос. Единственное, что тут можно ответить - такое использование возможно. Если возникнут конкретные вопросы, то оформляйте отдельные вопросы на Тостере.

[nuclear_skillet21011]

nirvimel: извини я зря отнял твоё время глупым вопросом(сказывается вторая бессонная ночь) спасибо за помощь.

[nuclear_skillet21011]

nirvimel: подскажите ещё пожалуйста: возможно ли определить какие из этих значений будут соседями для конкретной ячейки в матрице?

[nirvimel]

nuclear_skillet21011: У grid1[x, y] есть восемь соседних ячеек (в порядке по часовой стрелке): grid1[x-1, y-1], grid1[x, y-1], grid1[x+1, y-1], grid1[x+1, y], grid1[x+1, y+1], grid1[x, y+1], grid1[x-1, y+1], grid1[x-1, y]. Отдельно обрабатываются случаи, когда ячейка находится на краю массива.

[nuclear_skillet21011]

nirvimel:(извините, я уже наверно вам надоел) я пытаюсь реализовать то что вы описали посмотрите пожалуйста что я делаю не так:
у меня очень много вложенных циклов( наверняка есть более правильное решение) , в начале я создал список всех не нулевых значений, в следующих циклах я пытаюсь проверить по вашему(я думаю что по вашему) методу всех соседей для ячеек с ненулевым значением, но у меня просто зависает интерпретатор(из-за вложенности циклов надо думать) гляньте на код хоть одним глазком(код я положил в ответы к вопросу).

[nirvimel]

nuclear_skillet21011: Ну, решил я вам задачу (обновил ответ). Как расплачиваться будете? ;)

[nirvimel]

nuclear_skillet21011: Как оказывается, циклы там не нужны вообще!

[nuclear_skillet21011]

nirvimel: WM или ЯД ; ) ( да тут следует пояснить что я не студент, а вольнослушатель который учит программирование сам, возможности спросить у препода у меня нет)

[nuclear_skillet21011]

nirvimel: чёрт только вот в решении почти не чего не понимаю(пойду-ка я маны по numpy переводить)

Answer 2

[nuclear_skillet21011]

[ тот самый код ]

def count_neighbours(grid, row, col): # [row] - ряд
									  # [col] - столбец
	mass_el_1 = []
	grid1 = np.array(grid)

	mass8 = [(1,1),(1,2),(1,3),(2,1),(2,2),(2,3),(3,1),(3,2),(3,3)]

	mass3 = [(0,4),(4,4),(4,0),(0,0)]
	
	mass5 = [(1,0),(2,0),(3,0),(0,1),(0,2),(0,3),(1,4),(2,4),(3,4),(4,1),(4,2),(4,3)]
 
	xl = []
	yl = []
	for x in range(grid1.shape[0]):
		for y in range(grid1.shape[1]):
			if grid1[x, y] != 0:
				print (x, y)
				xl.append(x)
				yl.append(y)
				print(xl)
				print(yl)


	listindex = list(zip(xl,yl))
	print(listindex)

	xlll = []
	ylll = []

	neighbours = []
	exepts = []

	for x in range(grid1.shape[0]):
		for y in range(grid1.shape[1]):
			if grid1[x, y] != 0:
							
				for i in listindex:
					print('i: ',i)

					for i1 in mass8:
						print('i1: ',i1)

						if i == i1:
							
							try:
										
								if  grid1[x-1, y-1] == 1:
									neighbours.append((x-1, y-1))
								elif grid1[x, y-1] == 1:
									neighbours.append((x, y-1))
								elif grid1[x+1, y-1] == 1:
									neighbours.append((x+1, y-1))							
								elif grid1[x+1, y] == 1:
									neighbours.append((x+1, y))							
								elif grid1[x+1, y+1] == 1:
									neighbours.append((x+1, y+1))							
								elif grid1[x, y+1] == 1:
									neighbours.append((x, y+1))							
								elif grid1[x-1, y+1] == 1:
									neighbours.append((x-1, y+1))							
								elif grid1[x-1, y] == 1:
									neighbours.append((x-1, y))

							except IndexError :

								exepts.append((x,y))
							
	print(neighbours,"len:",len(neighbours))
	print(exepts,"len:",len(exepts))

count_neighbours(((1, 0, 0, 1, 0),
                  (0, 1, 0, 0, 0),
                  (0, 0, 1, 0, 1),
                  (1, 0, 0, 0, 0),
                  (0, 0, 1, 0, 0),),0,0)