тензорное расширение одномерного индекса в многомерный индекс - PullRequest
0 голосов
/ 13 сентября 2018

Я хочу сделать функцию многомерного массива из одномерного индексного массива с любым измерением.Например, у меня есть

x=np.array([1,2,3])
def extend_array(x,d)

Если я поставлю d = 2, то получающийся массив будет

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

, или вместо (x, d) можно ввести (n, d)) так что n ссылается на

np.array(range(1,n+1))

Я думаю, что для этого есть функция, но я не могу ее найти.Не могли бы вы помочь мне с этой проблемой?

1 Ответ

0 голосов
/ 13 сентября 2018
from itertools import product
t=list(product(x,repeat=d))

Это даст желаемый результат.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...