Как получить все координаты для n-dim массива, учитывая его размеры - PullRequest
Новая
       7

Как получить все координаты для n-dim массива, учитывая его размеры

0 голосов
/ 12 марта 2020

Допустим, вы получили кортеж для измерений второго массива, т. Е. (3,3) - это матрица 3x3, (4,5,6) - матрица 4x5x6 и т. Д. c. Как я могу написать функцию, которая может возвращать список всех возможных индексов? 

 dimensions = (2,2)
 get_coordinates(dimensions)
 >>[[0,0],[0,1],[1,0],[1,1]]

или 

 dimensions = (2,2,2)
 get_coordinates(dimensions)
 >>[[0,0,0],[0,1,0],[1,0,0],[1,1,0],[0,0,1],[0,1,1],[1,0,1],[1,1,1]]

1 Ответ

0 голосов
/ 12 марта 2020

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

def gen(t):
    if len(t) == 1:
        yield from range(t[0])

    else:   
        for e in range(t[0]):
            for g in gen(t[1:]):
                yield [e, *([g] if isinstance(g, int) else g)]

def get_coordinates(dim):
    return list(gen(dim))


print(get_coordinates((2, 2)))
print(get_coordinates((2, 2, 2)))

вывод: 

[[0, 0], [0, 1], [1, 0], [1, 1]]
[[0, 0, 0], [0, 0, 1], [0, 1, 0], [0, 1, 1], [1, 0, 0], [1, 0, 1], [1, 1, 0], [1, 1, 1]]
...