Получить все индексы N-мерного массива в виде списка

Есть ли способ быстро и эффективно получить список или массив всех индексов в N-мерном массиве в Python?

Например, image у нас есть такой массив:

import numpy as np

test = np.zeros((4,4))

array([[0., 0., 0., 0.],
       [0., 0., 0., 0.],
       [0., 0., 0., 0.],
       [0., 0., 0., 0.]])

Я хотел бы получить все индексы элементов следующим образом:

indices = [ [0,0],[0,1],[0,2] ... [3,2],[3,3] ]

Вы всегда можете использовать вложенные циклы for

— 20.09.2018 17:33

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

— 20.09.2018 17:34

Что-то в itertools должно уметь генерировать все комбинации.

— 20.09.2018 17:35

Буквально есть numpy.indices.

— 20.09.2018 17:38

Спасибо, numpy.indices показалось самым простым!

— 20.09.2018 17:55

python arrays numpy element indices

20.09.2018 17:31

Почему в Python есть оператор "pass"?

Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.

Некоторые методы, о которых вы не знали, что они существуют в Python

Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...

Основы Python Часть I

Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?

LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа

Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:

Оптимизация кода с помощью тернарного оператора Python

И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это

Советы по эффективной веб-разработке с помощью Python

Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.

109

Перейти к ответу Данный вопрос помечен как решенный

Ответы 5

Вы можете попробовать itertools.product:

>>> from itertools import product
>>> 
>>> [list(i) for i in product(range(4), range(4))]
[[0, 0], [0, 1], [0, 2], [0, 3], [1, 0], [1, 1], [1, 2], [1, 3], [2, 0], [2, 1], [2, 2], [2, 3], [3, 0], [3, 1], [3, 2], [3, 3]]

20.09.2018 17:37

Если вас устраивает понимание списка

test = np.zeros((4,4))
indices = [[i, j] for i in range(test.shape[0]) for j in range(test.shape[1])]
print (indices)

[[0, 0], [0, 1], [0, 2], [0, 3], [1, 0], [1, 1], [1, 2], [1, 3], [2, 0], [2, 1], [2, 2], [2, 3], [3, 0], [3, 1], [3, 2], [3, 3]]

20.09.2018 17:38

Я бы предложил сделать массив 1 той же формы, что и ваш массив test, используя np.ones_like, а затем используя np.where:

>>> np.stack(np.where(np.ones_like(test))).T
# Or np.dstack(np.where(np.ones_like(test)))
array([[0, 0],
       [0, 1],
       [0, 2],
       [0, 3],
       [1, 0],
       [1, 1],
       [1, 2],
       [1, 3],
       [2, 0],
       [2, 1],
       [2, 2],
       [2, 3],
       [3, 0],
       [3, 1],
       [3, 2],
       [3, 3]])

20.09.2018 17:40

Просто перечислить следует:

test = [[0., 0., 0., 0.],
       [0., 0., 0., 0.],
       [0., 0., 0., 0.],
       [0., 0., 0., 0.],
       [0., 0., 0., 0.]]

indices = [[i, j] for i, row in enumerate(test) for j, col in enumerate(row)]
print(indices)

>>> [[0, 0], [0, 1], [0, 2], [0, 3], [1, 0], [1, 1], [1, 2], [1, 3], [2, 0], [2, 1], [2, 2], [2, 3], [3, 0], [3, 1], [3, 2], [3, 3], [4, 0], [4, 1], [4, 2], [4, 3]]

Вы также можете преобразовать кортежи в список

— 20.09.2018 18:04

20.09.2018 17:44

Ответ принят как подходящий

Используйте np.indices с небольшим изменением формы:

np.indices(test.shape).reshape(2, -1).T

array([[0, 0],  
       [0, 1],  
       [0, 2],  
       [0, 3],  
       [1, 0],  
       [1, 1],  
       [1, 2],  
       [1, 3],  
       [2, 0],  
       [2, 1],  
       [2, 2],  
       [2, 3],  
       [3, 0],  
       [3, 1],  
       [3, 2],  
       [3, 3]])

20.09.2018 17:52

Другие вопросы по теме

Сортировка отрицательных элементов из массива

Свежий для java (массивы и простые числа), нужна помощь в устранении проблемы

Как получить недублирующий индекс массива в js?

Как использовать разные данные с помощью faker в Java с помощью функций?

Вычисление сумм и средних значений введенных чисел и сортировка чисел в Eclipse

Разделение маски на другие маски по ее элементам

Выделение массива размером более 2 ГБ в C (или C++)

Сортировка массива строк в соответствии с порядком другого массива с использованием первого столбца

Как массив типов значений размещается в куче?

Каждый n-й элемент из массива, начиная откуда-то в массиве C#

Получить все индексы N-мерного массива в виде списка

Ответы 5

Другие вопросы по теме

Похожие вопросы