Используя библиотеку numpy естественно. Т.к. у меня будут большие размеры перебор не подходит. Спрашивал на оверфлов и вот ответ один чел написал, но и это не ответ, т.к. при больший массивах например 200 на 354 с индексами 128 выдает ошибку в комментах оставлена там, исправить не удалось. Короче единственный способ который нашел просто используя np.where с заранее созданными массивами под индексы но это очень и очень медленно даже с оптимизацией по индексам.
