Должен ли я использовать словарь для хранения местоположения ненулевых пикселей?

Question

У меня есть двоичное изображение. Я хочу сохранить местоположение (x,y) всех ненулевых пикселей, чтобы использовать его для поиска в будущем.

import numpy as np
I = np.random.randint(2, size=(128,128))
pos_nonzero = np.where(I==1)

В будущем я случайно выберу позицию ненулевого пикселя для использования. Какой метод (словарь или список) я должен использовать здесь? Спасибо

Answer 1

Вы можете использовать np.argwhere для получения двумерного массива позиций, из которого вы можете пробовать позже, используя np.random.choice:

indices = np.argwhere(image)
sample_indices = np.random.choice(len(indices), size=5, replace=False)
pixels = image[tuple(indices[sample_indices].T)]